太長時間咸魚了！！

轉(zhuǎn)載：Cannot convert list to array: ValueError: only one element tensors can be converted to Python scalar

場景：我想將多個網(wǎng)絡輸出的結(jié)果（tensor類型）放到一個python list中， 然后直接轉(zhuǎn)換成numpy類型， 結(jié)果報錯

問題：只能將一個含有一個元素的Tensor轉(zhuǎn)換成python標量

解決辦法

對每一tensor先轉(zhuǎn)換成numpy類型，然后在進行操作

return torch.Tensor( np.array( [self.vgg16(item).numpy() for item in data] ) )

tensor -- > numpy? ? ? 方法是? ? .numpy()

numpy-->tensor? ? ? ? ? 方法是? ? torch.from_numpy()

實際操作：

feats = torch.Tensor(np.array([item.numpy() for item in feats]))