■

またまた、Pytorchの続き。
やりながら、結構Numpyに似てんなーと思ってたら、
案の定、Numpyとの連携もしてくれるみたい…

アウトプットです。

Pytorch → Numpy

import torch

a = torch.ones(5) <- tensor([1., 1., 1., 1., 1.])

b = a.numpy <- array([1., 1., 1., 1., 1.], dtype=float32)

Numpy -> Pytorch

a = np.ones(5) -> array([1., 1., 1., 1., 1.])

b = torch.from_numpy(a) -> tensor([1., 1., 1., 1., 1.], dtype=torch.float64)

np.add(a , 1 , out=a)
→このout=aが必要。

こうすると、NumpyとTorchの連携ができて、
お互いに数字の更新がなされるみたいです！