ValueError：在张量列表上使用 torch.Tensor 时，只能将一个元素张量转换为 Python 标量

Question

我有张量列表：

object_ids = [tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.])]

直觉上，我似乎应该能够从中创建一个新的张量：

torch.as_tensor(object_ids, dtype=torch.float32)

但这不起作用。显然，torch.as_tensor 和 torch.Tensor 只能将 标量 列表转换为新的张量。它不能将 d-dim 张量列表转换为 d+1 dim 张量。

Answer 1

您可以使用torch.stack。

在你的例子中：

>>> object_ids = [tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.]), tensor([2., 3.])]
>>> torch.stack(object_ids)
tensor([[2., 3.],
        [2., 3.],
        [2., 3.],
        [2., 3.],
        [2., 3.],
        [2., 3.],
        [2., 3.],
        [2., 3.],
        [2., 3.],
        [2., 3.]])