import torch
a = torch.tensor([[1,2,3], [4,5,6]])
a.transpose(0,1)
Out[5]:
tensor([[1, 4],
[2, 5],
[3, 6]])
torch tensor transpose函数:每次可以转置2 dim(线代上的转置)
最新推荐文章于 2024-02-20 11:53:28 发布
import torch
a = torch.tensor([[1,2,3], [4,5,6]])
a.transpose(0,1)
Out[5]:
tensor([[1, 4],
[2, 5],
[3, 6]])