Comment convertir une liste ou un tableau numpy en un tenseur de torche 1d?

Question

J'ai une liste (ou un tableau numérique) de valeurs flottantes. Je veux créer un tenseur de torche 1d qui contiendra toutes ces valeurs. Je peux créer le tenseur de la torche et exécuter une boucle pour stocker les valeurs.

Mais je veux savoir est-il possible, je peux créer un tenseur de la torche avec des valeurs initiales d'une liste ou d'un tableau? Suggérez-moi aussi s'il y a un moyen pythonique d'y parvenir puisque je travaille en pytorch.

Answer 1

Il s'agit d'opérations générales dans pytorch et disponibles dans la documentation. PyTorch permet une interface facile avec numpy. Il existe une méthode appelée from_numpy et la documentation est disponible here

import numpy as np 
import torch 
array = np.arange(1, 11) 
tensor = torch.from_numpy(array)