torch保存tensor

torch保存tensor可以通过torch.save()函数实现。torch.save()函数接收两个参数，第一个参数是要保存的tensor，可以是单个tensor或者是一个tensor列表，第二个参数是保存路径，可以是一个文件名或者是文件对象。

保存单个tensor的例子如下：

import torch

# 创建一个tensor
x = torch.tensor([1, 2, 3])

# 保存tensor到文件名为'x.pt'的文件中
torch.save(x, 'x.pt')

保存tensor列表的例子如下：

import torch

# 创建tensor列表
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5, 6])
z = torch.tensor([7, 8, 9])
tensor_list = [x, y, z]

# 保存tensor列表到文件名为'tensor_list.pt'的文件中
torch.save(tensor_list, 'tensor_list.pt')

保存后，可以使用torch.load()函数读取保存的tensor。torch.load()函数接收一个参数，即要加载的文件路径，返回一个保存的tensor或tensor列表。

读取保存的单个tensor的例子如下：

import torch

# 从文件名为'x.pt'的文件中加载tensor
x = torch.load('x.pt')

print(x)

读取保存的tensor列表的例子如下：

import torch

# 从文件名为'tensor_list.pt'的文件中加载tensor列表
tensor_list = torch.load('tensor_list.pt')

for tensor in tensor_list:
    print(tensor)

通过上述方法，就可以方便地保存和读取tensor。