import torch
a = torch.tensor([[1,2,3], [4,5,6]])
a.transpose(0,1)
Out[5]:
tensor([[1, 4],
[2, 5],
[3, 6]])
09-08
681
01-19
1371
import torch
a = torch.tensor([[1,2,3], [4,5,6]])
a.transpose(0,1)
Out[5]:
tensor([[1, 4],
[2, 5],
[3, 6]])