Función torch.load de PyTorch
Manual de referencia de Pytorch torch
torch.loadEs la función en PyTorch utilizada para la carga. Carga objetos serializados desde un archivo.
Definición de la función
torch.load(f, map_location=None, weights_only=False)
Ejemplo de uso
Ejemplo
import torch
# Guardar un tensor
x = torch.tensor([1, 2, 3, 4, 5])
torch.save(x, 'tensor.pt')
# Cargar tensor
loaded_x = torch.load('tensor.pt')
print(f"Loaded tensor: {loaded_x}")
# Guardar un tensor
x = torch.tensor([1, 2, 3, 4, 5])
torch.save(x, 'tensor.pt')
# Cargar tensor
loaded_x = torch.load('tensor.pt')
print(f"Loaded tensor: {loaded_x}")
Otras extensiones