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

Question

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

1x1、4x4、8x8 等

总共大约7个阵列。

我知道如何通过以下方式转换它们：

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

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

Answer 1

如果您所说的一个命令是指一行代码，那么在这里，我们可以使用列表理解lst = [a1by1, a4by4, a8by8] lst = [torch.from_numpy(item).float() for item in lst]