列表中的 numpy 數組是具有不同大小的二維數組，比方說：1x1、4x4、8x8 等總共大約7個陣列。我知道如何通過以下方式轉換它們：torch.from_numpy(a1by1).type(torch.FloatTensor)
torch.from_numpy(a4by4).type(torch.FloatTensor)
etc..有沒有一種方法可以通過一個命令轉換整個列表？
1 回答
胡子哥哥
TA貢獻1825條經驗 獲得超6個贊
如果您所說的一個命令是指一行代碼，那么
在這里，我們可以使用列表理解
lst = [a1by1, a4by4, a8by8] lst = [torch.from_numpy(item).float() for item in lst]
添加回答
舉報
0/150
提交
取消