Question

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

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

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

当我使用火炬张量（x）我得到一个错误 ValueError：只能将一个元素张量转换为Python标量

Answer 1

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相同。

Answer 2

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

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

还请注意，自right now起，您不能在cpu上使用带有半张量的cat，因此您应将它们转换为float，进行串联，然后转换回一半

state = { weatherData: { data: {} }, error: null };

您尚未说明目标，因此另一种选择是使用pad_sequence，如下所示：

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])

编辑：在这种情况下，您可以使用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])

将张量列表转换为张量pytorch

2 个答案: