numpy—array类型 与 pytorch—tensor类型 互相转换

一、numpy_array 转 torch_tensor

import torch

torch_data = torch.from_numpy(numpy_data)

二、torch_tensor 转 numpy_array

1 、 numpy_data = torch_data.numpy()

2、 import numpy as np

　　numpy_data = np.array(torch_data)