2018-10-04 Pytorch IndexError: too many indices for array

layout: post
title: IndexError: too many indices for array
subtitle: 一个由pytorch版本兼容性导致的问题
date: 2018-10-04
author: Johnny
header-img: img/post-bg-ios9-web.jpg
catalog: true
tags:
- iOS
- Demo

Pytorch 4.0 报错 IndexError: too many indices for array

出现问题的原因：pytorch 3.0 和 4.0以上版本不兼容，数据无法自动转换格式

action = action.data.numpy()[0].astype('int32')

改为

action = action.data.numpy()[0].astype('int32')

就可以了

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

推荐阅读更多精彩内容

  • 谈读书
    书，人类进步的阶梯。但，这阶梯我们真的了解吗？ 每位孩童自幼便开始读书，教科书、童话书、故事书等他们自认为对他...
    c雪初阅读 454评论 7赞 7
  • 【鹧鸪天】天山天池
    雪海三峰插云天，飞瀑泻玉半空悬。 珍珠流泉歌声醉，闲云竞姿映碧潭。 瀑轰鸣，涌雾岚。彩练当空气万千。 水烟缥缈隐千...
    此木可阅读 463评论 0赞 0
  • 做你自己 我来爱你
    要我成熟给谁看阅读 141评论 0赞 0