将 numpy 数组列表转换为火炬张量列表

Question

列表中的 numpy 数组是具有不同大小的二维数组，比方说：

1x1、4x4、8x8 等

总共大约7个数组。

我知道如何转换它们中的每一个，方法是：

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

有没有办法在一个命令中转换整个列表？

我发现了这 2 个问题：

如何将列表或 numpy 数组转换为一维 torch 张量？

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

但这不是我要找的东西

Answer 1

如果通过一个命令你的意思是 one-liner 那么

这里，我们可以使用list-comprehension

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