1.torch.tensor 轉換爲numpy:
a = torch.ones(5)
b = a.data.numpy()
2.numpy轉換爲torch.tensor:
import numpy as np a = np.ones(5) b = torch.from_numpy(a) np.add(a, 1, out=a)
pytorch編程
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
1.torch.tensor 轉換爲numpy:
a = torch.ones(5)
b = a.data.numpy()
2.numpy轉換爲torch.tensor:
import numpy as np a = np.ones(5) b = torch.from_numpy(a) np.add(a, 1, out=a)