【pytorch】numpyからtensor

pytorchでnumpyからtensorに変える方法を書く。

numpyからtensor

torch.from_numpy(ndarray)をつかう。
ndarrayにはnumpyの行列が入る。

>>>import numpy
>>>import torch
>>> a = numpy.array([0, 1, 2])
>>> t = torch.from_numpy(a)
>>> t
tensor([ 0, 1,  2])
>>> t[0] = 1
>>> a
array([ 1,  1,  2])

上の例から、arrayをtensorに変えることができていることが分かる。
また、aのarrayとtのtensorはメモリを共有していることが分かる。