Écrivez comment passer de numpy à tenseur avec pytorch.
Utilisez torch.from_numpy (ndarray). Le ndarray contient une matrice 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])
À partir de l'exemple ci-dessus, vous pouvez voir que le tableau peut être changé en tenseur. On voit aussi que le tableau de a et le tenseur de t partagent la mémoire.
Recommended Posts