Función PyTorch torch.frombuffer
Manual de referencia de Pytorch torch
torch.frombufferEs una función en PyTorch que se utiliza para crear un tensor unidimensional a partir de un buffer.
Definición de la función
torch.frombuffer(buffer, dtype, count, offset)
Ejemplo de uso
Ejemplo
import torch
import array
# Crear desde array
arr = array.array('i', [1, 2, 3, 4, 5])
x = torch.frombuffer(arr, dtype=torch.int32)
print(x)
import array
# Crear desde array
arr = array.array('i', [1, 2, 3, 4, 5])
x = torch.frombuffer(arr, dtype=torch.int32)
print(x)
Otras extensiones