pytorch每日一学16(torch.from_numpy())将numpy转化为tensor

第16个方法

torch.from_numpy(ndarray)

• 此方法非常的直观，就是将一个numpy.ndarray转化为tensor。
• 注意，返回的tensor和原来的ndarray共享内存，更改tensor也会更改ndarray。并且返回的tensor不能调整大小。

目前能接受的ndarrays类型有numpy.float64, numpy.float32, numpy.float16, numpy.complex64, numpy.complex128, numpy.int64, numpy.int32, numpy.int16, numpy.int8, numpy.uint8, and numpy.bool.

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