笔记68:Pytorch中repeat函数的用法

repeat 相当于一个broadcasting的机制

repeat(*sizes)

沿着指定的维度重复tensor。不同与expand(),本函数复制的是tensor中的数据。

python 复制代码
import torch
import torch.nn.functional as F
import numpy as np
a = torch.Tensor(128,1,512)
B = a.repeat(1,5,1)
print(B.shape)
python 复制代码
torch.Size([128, 5, 512])

转自:pytorch repeat的用法-CSDN博客

相关推荐
三品吉他手会点灯7 小时前
C语言学习笔记 - 12.C语言简介 - 一元二次方程详解
c语言·笔记·学习
Cathy Bryant8 小时前
微分几何:度规(度量)metric
笔记·线性代数·矩阵·高等数学·物理
fengxin_rou9 小时前
JVM 核心笔记:对象创建、生命周期与类加载器详解
java·jvm·笔记
weixin_704266059 小时前
Spring Cloud Gateway 完整版笔记
笔记
qeen8710 小时前
【算法笔记】二分查找与二分答案
c语言·c++·笔记·学习·算法·二分
脆皮炸鸡75510 小时前
进程的程序替换
linux·经验分享·笔记·vim·学习方法
萌新小码农‍10 小时前
机器学习概述 学习笔记day2
笔记·学习·机器学习
daanpdf10 小时前
大一《中国近代史纲要》题库及答案PDF知识点整理笔记
笔记·pdf
曦月逸霜10 小时前
区块链技术与应用学习笔记(持续更新中)
笔记·学习·区块链
中屹指纹浏览器12 小时前
跨终端跨平台指纹浏览器的环境一致性技术架构与落地实践
经验分享·笔记