温馨提示×

pytorch中如何保存tensor数据

小亿
141
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