pytorch张量与列表之间的转换

在PyTorch中，可以使用torch.Tensor()函数创建张量。如果要将一个列表转换成张量，可以使用torch.tensor()函数。下面是一个例子：

import torch

# 创建一个列表
my_list = [1, 2, 3, 4, 5]

# 将列表转换成张量
my_tensor = torch.tensor(my_list)

print(my_tensor)

输出：

tensor([1, 2, 3, 4, 5])

如果要将一个张量转换成列表，可以使用张量的tolist()方法。下面是一个例子：

import torch

# 创建一个张量
my_tensor = torch.tensor([1, 2, 3, 4, 5])

# 将张量转换成列表
my_list = my_tensor.tolist()

print(my_list)

输出：

[1, 2, 3, 4, 5]