参数的size要求和输入向量一致,若TensorA有三维则
TensorA.repeat(?,?,?)代表各自维度重复多少次
A = torch.tensor([[[0, 0],
[1, 1],
[2, 2]]])
现在重复列2次,两列变四列


现在重复行2次,三行变六列


同样,现在一个3x2的变成两个3x2的


Pytorch中的repeat()函数
阅读 83
2022-04-17
参数的size要求和输入向量一致,若TensorA有三维则
TensorA.repeat(?,?,?)代表各自维度重复多少次
A = torch.tensor([[[0, 0],
[1, 1],
[2, 2]]])
现在重复列2次,两列变四列


现在重复行2次,三行变六列


同样,现在一个3x2的变成两个3x2的


相关推荐
精彩评论(0)