参数的size要求和输入向量一致,若TensorA有三维则
TensorA.repeat(?,?,?)代表各自维度重复多少次
A = torch.tensor([[[0, 0],
[1, 1],
[2, 2]]])
现在重复列2次,两列变四列
现在重复行2次,三行变六列
同样,现在一个3x2的变成两个3x2的
微信扫一扫
参数的size要求和输入向量一致,若TensorA有三维则
TensorA.repeat(?,?,?)代表各自维度重复多少次
A = torch.tensor([[[0, 0],
[1, 1],
[2, 2]]])
现在重复列2次,两列变四列
现在重复行2次,三行变六列
同样,现在一个3x2的变成两个3x2的
相关推荐