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 tensor: {loaded_x}")
# 保存一个张量
x = torch.tensor([1, 2, 3, 4, 5])
torch.save(x, 'tensor.pt')
# 加载张量
loaded_x = torch.load('tensor.pt')
print(f"Loaded tensor: {loaded_x}")