pytorch 中repeat函数

沿着指定的维度重复tensor。不同与expand(),本函数复制的是tensor中的数据。
扩展(expand)张量不会分配新的内存,只是在存在的张量上创建一个新的视图(view),一个大小(size)等于1的维度扩展到更大的尺寸。repeat沿着特定的维度重复这个张量,和expand()不同的是,这个函数拷贝张量的数据。

import torch
a=torch.linspace(0,12,13)
print(a)
print(a.shape)

pytorch 中repeat函数

b=a.repeat(13,1)
print(b)
print(b.shape)

pytorch 中repeat函数

c=b.repeat(3,1,1)
print(c)
print(c.shape)

pytorch 中repeat函数

上一篇:IOS开发-经常使用站点集合


下一篇:整理一下linux系统expand 命令