Schreiben Sie, wie Sie mit Pytorch von Numpy zu Tensor wechseln.
Verwenden Sie torch.from_numpy (ndarray). Das ndarray enthält eine Numpy-Matrix.
>>>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])
Aus dem obigen Beispiel können Sie ersehen, dass das Array in Tensor geändert werden kann. Es ist auch ersichtlich, dass sich das Array von a und der Tensor von t den Speicher teilen.
Recommended Posts