PyTorch torch.load Function
PyTorch torch Reference Manual
torch.loadis a function in PyTorch used for loading. Loads serialized objects from a file.
Function Definition
torch.load(f, map_location=None, weights_only=False)
Usage Examples
Example
import torch
# Save a tensor
x = torch.tensor([1, 2, 3, 4, 5])
torch.save(x, 'tensor.pt')
# Load tensor
loaded_x = torch.load('tensor.pt')
print(f"Loaded tensor: {loaded_x}")
# Save a tensor
x = torch.tensor([1, 2, 3, 4, 5])
torch.save(x, 'tensor.pt')
# Load tensor
loaded_x = torch.load('tensor.pt')
print(f"Loaded tensor: {loaded_x}")
Other Extensions