PyTorch torch.load Функция
Справочное руководство PyTorch torch
torch.loadявляется функцией в PyTorch, используемой для загрузки. Загружает сериализованные объекты из файла.
Определение функции
torch.load(f, map_location=None, weights_only=False)
Примеры использования
Пример
import torch
# Сохранить тензор
x = torch.tensor([1, 2, 3, 4, 5])
torch.save(x, 'tensor.pt')
# Загрузить тензор
loaded_x = torch.load('tensor.pt')
print(f"Загруженный тензор: {loaded_x}")
# Сохранить тензор
x = torch.tensor([1, 2, 3, 4, 5])
torch.save(x, 'tensor.pt')
# Загрузить тензор
loaded_x = torch.load('tensor.pt')
print(f"Загруженный тензор: {loaded_x}")
Другие расширения