溫馨提示×

pytorch中如何保存tensor數據

小億
709
2024-03-27 14:20:29
欄目: 深度學習

在PyTorch中,可以使用torch.save()函數將Tensor數據保存到文件中。以下是保存和加載Tensor數據的示例代碼:

import torch

# 創建一個Tensor數據
tensor_data = torch.tensor([[1, 2], [3, 4]])

# 保存Tensor數據到文件
torch.save(tensor_data, 'tensor_data.pt')

# 加載保存的Tensor數據
loaded_tensor_data = torch.load('tensor_data.pt')

print(loaded_tensor_data)

在上面的示例中,我們首先創建一個包含Tensor數據的變量tensor_data,然后使用torch.save()函數將這個Tensor數據保存到名為tensor_data.pt的文件中。接著我們使用torch.load()函數加載保存的Tensor數據,并將其保存到loaded_tensor_data變量中。最后打印加載的Tensor數據。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女