首页 Paddle框架 帖子详情
paddle试图索引失败
收藏
快速回复
Paddle框架 问答模型训练 454 1
paddle试图索引失败
收藏
快速回复
Paddle框架 问答模型训练 454 1

张量A的规模是(64,8,32,32)

但我试图索引张量A,即A[0, [2,3,1,0], :, :]时,发生错误

ValueError: (InvalidArgument) When index contains a list, its length is excepted to 1, but received 4
[Hint: Expected size == 1, but received size:4 != 1:1.] (at /paddle/paddle/fluid/pybind/imperative.cc:599)

请问该如何解决,同样的代码在torch上是可以跑通的

0
收藏
回复
全部评论(1)
时间顺序
十进制到二进制
#2 回复于2022-03

paddle应该不支持这种切片格式,还是一张张取,然后concat起来吧

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户