paddle试图索引失败
收藏
张量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
收藏
请登录后评论
paddle应该不支持这种切片格式,还是一张张取,然后concat起来吧