Fonction torch.as_tensor de PyTorch
Manuel de référence Pytorch torch
torch.as_tensorest une fonction dans PyTorch utilisée pour convertir des données en tenseurs. Contrairement àtorch.tensoril partage la mémoire des données autant que possible.
Définition de fonction
torch.as_tensor(data, dtype=None, device=None)
Exemple d'utilisation
Exemple
import torch
import numpy as np
# Créer à partir d'un tableau NumPy (mémoire partagée)
numpy_array = np.array([1, 2, 3])
x = torch.as_tensor(numpy_array)
# Modifier le tableau NumPy
numpy_array[0] = 100
print("Tensor:", x) # Le tenseur changera aussi
import numpy as np
# Créer à partir d'un tableau NumPy (mémoire partagée)
numpy_array = np.array([1, 2, 3])
x = torch.as_tensor(numpy_array)
# Modifier le tableau NumPy
numpy_array[0] = 100
print("Tensor:", x) # Le tenseur changera aussi
Le résultat de sortie est:
张量: tensor([100, 2, 3])
Autres extensions