pytorch中如何保存tensor数据

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