PyTorch torch.load 函数

PyTorch torch.load 函数 Pytorch torch 参考手册 torch.load 是 PyTorch 中用于加载的函数。从文件中加载序列化的对象。函数定义 torch.load(f, map_location=None, weights_only=False) 使用示例实例 [mycode4 type='python'] import torch # 保存一个张量 x = to..

更新于 2026/7/16

教程详情页默认 noindex。

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}")

Pytorch torch 参考手册