如何将含tensor数据类型的list转换为tensor

可以使用torch.stack()函数将含有tensor数据类型的list转换为tensor。torch.stack()函数的作用是将一个list中的tensor沿着指定的维度进行堆叠，返回一个新的tensor。

示例代码如下：

import torch

# 定义一个含有tensor数据类型的list
tensor_list = [torch.tensor([1, 2]), torch.tensor([3, 4]), torch.tensor([5, 6])]

# 使用torch.stack()函数将list转换为tensor
tensor = torch.stack(tensor_list, dim=0)  # 沿着第0维进行堆叠

print(tensor)

输出：

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

在上面的示例代码中，我们定义了一个含有tensor数据类型的list，然后使用torch.stack()函数将list转换为tensor。在函数调用中，我们指定了dim=0，表示沿着第0维进行堆叠。最终输出的结果是一个2维的tensor，其中每一行都是原来list中的一个tensor。