专栏推荐
本文重点
将tensor转成numpy
import torch
import numpy as np
data=np.array([1,2,3])
b=torch.from_numpy(data)
print(b)
将numpy转成tensor
n=np.random.randn(10,20)#定义一个numpy类型
tensor1 = torch.Tensor(n)#方式一：numpy转tensor
tensor2 = torch.from_numpy(n)#方式二：numpy转tensor
n1= tensor2.numpy()#tensor转numpy
#numpy_array = pytorch_tensor1.cpu().numpy()#tensor转numpy(tensor在GPU上的话需要先转到cpu)
#print(n1)
e=d.numpy()
print (e)#可以直接使用点numpy的方式来将tensor类型的来与numpy的ndarray之间进行转换
f=torch.from_numpy(e)#可以使用from_numpy的方式来将numpy转成tensor
print (f)
本文来自投稿，不代表本人立场，如若转载，请注明出处：http://www.sosokankan.com/article/1928837.html
赞 (1)
打赏
中国70%的手机都用北斗导航，就剩下iPhone这块硬骨头了
« 上一篇2019-12-31 15:11:18
华为手机自带会议神器！真没想到，居然这么好用，录音笔都用不着
下一篇 »2019-12-31 15:21:00