import torch
import numpy as np
# reshape是将原始的一维数据重新整合,不涉及到维度的转换,但是在permute涉及到转置
np_data = np.arange(0,6)
np_data_ = torch.from_numpy(np_data).reshape(2,3)
print(np_data_.reshape(3,2))
print(np_data_.permute(1,0))
print(torch.transpose(np_data_, 1,0)) #注意,一次只能交换两个维度
result:
tensor([[0, 1],
[2, 3],
[4, 5]])
tensor([[0, 3],
[1, 4],
[2, 5]])
tensor([[0, 3],
[1, 4],
[2, 5]])
torch中permutetranspose reshape
最新推荐文章于 2024-02-17 11:23:52 发布