将numpy数组的列表转换为torch张量列表

Question

列表中的numpy数组是具有不同大小的2D数组，例如：

1x1、4x4、8x8等

总共约7个阵列。

我知道如何转换它们中的每一个，通过：

torch.from_numpy(a1by1).type(torch.FloatTensor)
torch.from_numpy(a4by4).type(torch.FloatTensor)
etc..

有没有一种方法可以在一个命令中转换整个列表？

我发现这两个问题：

如何将列表或numpy数组转换为1d火炬张量？

如何将张量列表转换为torch:：张量？

但这不是我想要的

Answer 1

如果您所说的一个命令是指一个liner，那么

在这里，我们可以使用列表理解

lst = [a1by1, a4by4, a8by8]
lst = [torch.from_numpy(item).float() for item in lst]