pytorch中如何保存tensor数据

pytorch

604

2024/3/28 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数据。

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

pytorch中如何保存tensor数据

最新知识库

相关标签