web-dev-qa-db-fra.com

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

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

Mais je veux savoir s'il existe un moyen, je peux créer un tenseur de torche avec des valeurs initiales à partir d'une liste ou d'un tableau? Suggérez-moi également s'il existe un moyen Pythonique pour y parvenir car je travaille dans Pytorch.

11
Wasi Ahmad

Ce sont des 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 ici

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