在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数据。