PyTorch torch.load 関数
Pytorch torch リファレンスマニュアルtorch.loadPyTorch でロードに使用される関数です。ファイルからシリアライズされたオブジェクトをロードします。
関数定義
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}")
その他の拡張