Fonction PyTorch torch.frombuffer
Manuel de référence PyTorch torch
torch.frombufferC'est une fonction de PyTorch utilisée pour créer un tenseur unidimensionnel à partir d'un tampon.
Définition de fonction
torch.frombuffer(buffer, dtype, count, offset)
Exemple d'utilisation
Exemple
import torch
import array
# Créer à partir d'un tableau
arr = array.array('i', [1, 2, 3, 4, 5])
x = torch.frombuffer(arr, dtype=torch.int32)
print(x)
import array
# Créer à partir d'un tableau
arr = array.array('i', [1, 2, 3, 4, 5])
x = torch.frombuffer(arr, dtype=torch.int32)
print(x)
Autres extensions