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

时间:2020-10-19 22:27:53

标签: python list numpy pytorch torch

列表中的numpy数组是具有不同大小的2D数组,假设:

1x1、4x4、8x8等

总共约7个阵列。

我知道如何通过以下方法转换它们中的每一个:

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

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

我发现了这两个问题:

How to convert a list or numpy array to a 1d torch tensor?

How to convert a list of tensors into a torch::Tensor?

但这不是我要寻找的

1 个答案:

答案 0 :(得分:1)

如果用一条命令来表示一线那么

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

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