pytorch

pytorch张量转置方法是什么

小樊
81
2024-12-25 20:56:25
栏目: 深度学习

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

0
看了该问题的人还看了