在PyTorch中,可以使用.T
或.transpose()
方法对张量进行转置。
例如,假设有一个二维张量x
:
import torch x = torch.tensor([[1, 2], [3, 4]]) print(x)
输出:
tensor([[1., 2.], [3., 4.]])
使用.T
方法进行转置:
x_transposed = x.T print(x_transposed)
输出:
tensor([[1., 3.], [2., 4.]])
使用.transpose()
方法进行转置,可以指定轴数:
x_transposed = x.transpose(0, 1) print(x_transposed)
输出:
tensor([[1., 3.], [2., 4.]])
在这个例子中,transpose()
方法的参数0, 1
表示将第一个轴和第二个轴互换位置。