![7ffc7aa19a7325252af4804845f3fa80.png](https://i-blog.csdnimg.cn/blog_migrate/46f6f9c7cbb0ea9fcca4691f3ec9369b.jpeg)
****************************************本文参考****************************************
Torch 或 Numpy - PyTorch | 莫烦Pythonmofanpy.com![e2893b7b59ff375dfef9e4d567b87abc.png](https://i-blog.csdnimg.cn/blog_migrate/e5e0c2306dfa2ea93a52d9c60c2d5cd1.png)
*****************************************************************************************
1 Numpy array 和 Torch tensor 的转换
【1】
import torch
import numpy as np
np_data = np.arange(6).reshape((2, 3))
torch_data = torch.from_numpy(np_data)
tensor2array = torch_data.numpy()
print(
'nnumpy array:', np_data,
'ntorch tensor:', torch_data,
'ntensor to array:'