pytorchプログラミング

1.torch.tensorをnumpyに変換します。

a = torch.ones(5)

b = a.data.numpy()

2.numpyをtorch.tensorに変換します。

numpyを
npa = np.ones(5)
b = torch.from_numpy(a)
np.add(a、1、out = a)としてインポートします

 

おすすめ

転載: blog.csdn.net/sinat_39307513/article/details/86099420