import torch
torch_a=torch.tensor([[2,2,3],[4,4,5]])
numpy_a=torch_a.data.numpy()
numpy_a_t=numpy_a.transpose()
torch_a_t01=torch_a.transpose(0,1)
numpy_a_t_tensor=torch.from_numpy(numpy_a_t)
torch_a=torch.tensor([[2,2,3],[4,4,5]])
numpy_a_t:
shape:(3, 2)
torch_a_t01:tensor([[2, 4],
[2, 4],
[3, 5]])
shape:torch.Size([3, 2])
numpy_a_t_tensor:tensor([[2, 4],
[2, 4],
[3, 5]])
shape:torch.Size([3, 2])
到了要保存mat文件的时候。直接使用numpy操作,mat文件数据为double型的。若用tenor操作,则mat文件数据为single型的。比double的数据大小小了一半。