在PyTorch中,可以使用torch.t()函数来对矩阵进行转置。下面是一个示例代码:

import torch

# 创建一个2x3的矩阵
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 对矩阵进行转置
x_transposed = torch.t(x)

print(x_transposed)

输出结果为:

tensor([[1, 4],
        [2, 5],
        [3, 6]])

可以看到,原矩阵的行变成了转置矩阵的列,原矩阵的列变成了转置矩阵的行。


原文地址: https://www.cveoy.top/t/topic/pzsy 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录