标签:相互 and tor pre div 相互转换 import bsp print
import torch x = torch.rand(2,2) x1 = x.numpy() # torch转换到numpy x2 = torch.from_numpy(x1) #numpy转换torch print("\n torch_x:\n",x, "\n numpy_x1:\n",x1, "\n torch_x2:\n",x2)
out:
torch_x: tensor([[0.8195, 0.8927], [0.7544, 0.8827]]) numpy_x1: [[0.8194596 0.8926985] [0.754393 0.882745 ]] torch_x2: tensor([[0.8195, 0.8927], [0.7544, 0.8827]])
标签:相互 and tor pre div 相互转换 import bsp print
原文地址:https://www.cnblogs.com/peixu/p/13208718.html