PyTorch矩阵转置:使用torch.t()函数进行转置操作
在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 著作权归作者所有。请勿转载和采集!