pytorch和Numpy的区别以及相互转换
·
numpy中的ndarray转化成pytorch中的tensor :
torch.from_numpy()
pytorch中的tensor转化成numpy中的ndarray :
numpy()
import numpy as np
import torch
np_arr = np.array([1,2,3,4])
tor_arr=torch.from_numpy(np_arr)
tor2numpy=tor_arr.numpy()
print('\nnumpy\n',np_arr,'\ntorch\n',tor_arr,'\nnumpy\n',tor2numpy)
更多推荐



所有评论(0)