将张量列表转换为张量 pytorch

Question

我有张量列表，每个张量都有不同的大小如何使用 pytroch 将此张量列表转换为张量

有关更多信息，我的列表包含张量，每个张量都有不同的大小 例如第一个张量大小是 torch.Size([76080, 38])

其他张量的形状在第二个元素中会有所不同，例如列表中的第二个张量是 torch.Size([76080, 36])

当我使用 火炬.张量（x） 我收到一个错误 ValueError: 只有一个元素张量可以转换为 Python 标量

Answer 1

张量不能保存可变长度的数据。你可能正在寻找cat

例如，这里我们有一个列表，其中包含两个具有不同大小的张量（在它们的最后一个 dim(dim=2) 中），我们想要创建一个由它们组成的更大的张量，因此我们可以使用 cat 并创建一个更大的张量，包含他们的两个数据。

还请注意，从 right now 开始，您不能在 cpu 上使用 cat 和半张量，因此您应该将它们转换为浮点数，进行连接，然后再转换回半张量

import torch

a = torch.arange(8).reshape(2, 2, 2)
b = torch.arange(12).reshape(2, 2, 3)
my_list = [a, b]
my_tensor = torch.cat([a, b], dim=2)
print(my_tensor.shape) #torch.Size([2, 2, 5])

你还没有解释你的目标，所以另一个选择是像这样使用pad_sequence：

from torch.nn.utils.rnn import pad_sequence
a = torch.ones(25, 300)
b = torch.ones(22, 300)
c = torch.ones(15, 300)
pad_sequence([a, b, c]).size() #torch.Size([25, 3, 300])

编辑：在这种特殊情况下，您可以使用torch.cat([x.float() for x in sequence], dim=1).half()

Answer 2

pytorch 中的Tensor 与python 中的List 不同，后者可以保存可变长度的对象。

在pytorch中，可以将一个固定长度的数组传递给Tensor：

>>> torch.Tensor([[1, 2], [3, 4]])
>>> tensor([[1., 2.],
            [3., 4.]])

而不是：

>>> torch.Tensor([[1, 2], [3, 4, 5]])
>>> 
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-16-809c707011cc> in <module>
----> 1 torch.Tensor([[1, 2], [3, 4, 5]])

ValueError: expected sequence of length 2 at dim 1 (got 3)

和torch.stack一样。