Функция PyTorch torch.frombuffer
Справочное руководство PyTorch torch
torch.frombufferЭто функция в PyTorch, используемая для создания одномерного тензора из буфера.
Определение функции
torch.frombuffer(buffer, dtype, count, offset)
Пример использования
Пример
import torch
import array
# Создать из array
arr = array.array('i', [1, 2, 3, 4, 5])
x = torch.frombuffer(arr, dtype=torch.int32)
print(x)
import array
# Создать из array
arr = array.array('i', [1, 2, 3, 4, 5])
x = torch.frombuffer(arr, dtype=torch.int32)
print(x)
Другие расширения