pytorch 中repeat函数

沿着指定的维度重复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)

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

标签：repeat,函数,shape,张量,维度,pytorch,print,expand
来源： https://blog.csdn.net/weixin_39289876/article/details/116746557