在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
表示将第一个轴和第二个轴互换位置。