在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數據。