参数的size要求和输入向量一致,若TensorA有三维则
TensorA.repeat(?,?,?)代表各自维度重复多少次
A = torch.tensor([[[0, 0],
[1, 1],
[2, 2]]])
现在重复列2次,两列变四列
现在重复行2次,三行变六列
同样,现在一个3x2的变成两个3x2的
Pytorch中的repeat()函数
阅读 81
2022-04-17
参数的size要求和输入向量一致,若TensorA有三维则
TensorA.repeat(?,?,?)代表各自维度重复多少次
A = torch.tensor([[[0, 0],
[1, 1],
[2, 2]]])
现在重复列2次,两列变四列
现在重复行2次,三行变六列
同样,现在一个3x2的变成两个3x2的
相关推荐
精彩评论(0)