pytorch保存张量

在PyTorch中，可以使用torch.save()函数将张量保存到文件中。以下是保存张量的步骤：

1. 首先，确保已经导入了PyTorch库：

import torch

2. 创建一个张量并赋予它一些值：

tensor = torch.tensor([1, 2, 3, 4, 5])

3. 使用torch.save()函数将张量保存到文件中。需要提供两个参数：要保存的张量和保存的文件路径。文件路径可以是相对路径或绝对路径。

torch.save(tensor, 'tensor.pt')

4. 这样，张量就会被保存到名为'tensor.pt'的文件中。