pytorch(6)tensor_reshape()和tensor_transpose()函数

60 阅读 0 评论 40 点赞

我是靠谱客的博主花痴早晨，这篇文章主要介绍pytorch(6)tensor_reshape()和tensor_transpose()函数，现在分享给大家，希望可以做个参考。

tensor_reshape()将原始张量修改为指定形状大小的张量。
tensor_transpose()函数将原始张量的维度进行调换。

import torch
import numpy as np
def tensor_reshape():
arr = np.array([[[1, 2, 3], [4, 5, 6], [7, 8, 9]],
[[1, 2, 3], [4, 5, 6], [7, 8, 9]],
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]])
t = torch.tensor(arr)
t1 = torch.reshape(t,(-1,1))
print(t1.shape)
def tensor_transpose():
arr = np.array([[[1, 2, 3,4], [4, 5, 6,7], [7, 8, 9,10]],
[[1, 2, 3,4], [4, 5, 6,7], [7, 8, 9,10]],
[[1, 2, 3,4], [4, 5, 6,7], [7, 8, 9,10]]])
print(arr)
t = torch.tensor(arr)
t1 = torch.transpose(t,0,2)
print(t1,t1.shape)
if __name__ == '__main__':
# tensor_reshape()
tensor_transpose()