沿着指定的维度重复tensor。不同与expand(),本函数复制的是tensor中的数据。
扩展(expand)张量不会分配新的内存,只是在存在的张量上创建一个新的视图(view),一个大小(size)等于1的维度扩展到更大的尺寸。repeat沿着特定的维度重复这个张量,和expand()不同的是,这个函数拷贝张量的数据。
import torch
a=torch.linspace(0,12,13)
print(a)
print(a.shape)
文章图片
b=a.repeat(13,1)
print(b)
print(b.shape)
【pytorch 中repeat函数】
文章图片
c=b.repeat(3,1,1)
print(c)
print(c.shape)
文章图片
推荐阅读
- Pytorch|Pytorch丨expand函数
- pytorch|Pytorch函数expand()详解
- pytorch|pytorch中expand()和repeat()的区别
- 李宏毅机器学习|李宏毅2021年机器学习作业2(classification)实验记录
- 深度学习|基于pytorch用yolov5算法实现目标检测与分割
- 深度学习|在cmd中可以import torch在jupyter notebook却no module name “torch“
- pytorch学习笔记|pytorch-resnet34残差网络理解
- PyTorch|PyTorch: hook机制
- 动手学习深度学习|《动手学深度学习》Task04(机器翻译及相关技术+注意力机制与Seq2seq模型+Transformer)