pytorch tensor按维度复制

可以使用PyTorch中的repeat函数来按维度复制tensor。repeat函数接受一个参数，表示每个维度需要复制的次数，可以传入一个元组来指定每个维度的复制次数。例如，对于一个形状为(3,4)的tensor，可以使用以下代码按行复制两次：

import torch

x = torch.randn(3, 4)
y = x.repeat(2, 1)
print(y.shape)  # 输出(6, 4)

这里的(2,1)表示第一维复制2次，第二维复制1次。如果要按列复制两次，可以使用(1,2)作为参数：

z = x.repeat(1, 2)
print(z.shape)  # 输出(3, 8)

这里的(1,2)表示第一维复制1次，第二维复制2次。