服务器系列(22):Cannot convert list to array: ValueError: only one element tensors can be converted to ...

太长时间咸鱼了!!

转载:Cannot convert list to array: ValueError: only one element tensors can be converted to Python scalar

场景:我想将多个网络输出的结果(tensor类型)放到一个python list中, 然后直接转换成numpy类型, 结果报错

问题:只能将一个含有一个元素的Tensor转换成python标量

解决办法

对每一tensor先转换成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]))

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 什么是Tensor Tensor,又名张量,最早接触这个词是来自于TensorFlow,这个概念也被广泛的应用于不...
    Daisy丶阅读 54,085评论 1 20
  • 概述 在新版本中,PyTorch引入了许多令人兴奋的新特性,主要的更新在于 Variable和Tensor的合并 ...
    古de莫宁阅读 6,185评论 0 1
  • 在使用电脑的时候经常会用到各种功能的软件,有的软件上手难度大,需要在网上看软件操作教程视频才能看懂,用迅捷屏幕录像...
    一个石头哦阅读 1,150评论 0 0
  • 初见, 你我,本就是怨念一场, 谁料,且愿有几箱。 烟暖雨上,白缕飘渺。 闲云高山流水间, 你不经意的出场, 驻进...
    寒兮_6962阅读 121评论 1 1
  • 山巔青松勁, 霧籠蒼穹間。 登高望遠眺, 人世紅塵煙。
    行者江山阅读 107评论 0 0