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