Paddle 关于使用array来索引tensor

pxyaymoc 于 2021-12-07 发布在 Java

关注(0)|答案(9)|浏览(389)

这是UCN的pytorch版本，x和y是特征点的坐标，源码就是这样索引的。然后在paddle下还没有这样的语法，请问paddle有没有增加这个功能的准备，多久可以支持使用array来索引tensor。

Paddle

来源：https://github.com/PaddlePaddle/Paddle/issues/33605

9条答案

按热度按时间

ttvkxqim1#

您好，我们已经收到了您的问题，会安排技术人员尽快解答您的问题，请耐心等待。请您再次检查是否提供了清晰的问题描述、复现代码、环境&版本、报错信息等。同时，您也可以通过查看官网API文档、常见问题、历史Issue、AI社区来寻求解答。祝您生活愉快～

Hi! We've received your issue and please be patient to get responded. We will arrange technicians to answer your questions as soon as possible. Please make sure that you have posted enough message to demo your request. You may also check out the API，FAQ，Github Issue and AI community to get the answer.Have a nice day!

赞(0）回复(0）举报 2021-12-07

nnt7mjpx2#

hi, paddle最近也在支持比较丰富的slice index类型，以array作为index的，可以参考这个PR: #33000

赞(0）回复(0）举报 2021-12-07

ev7lccsx3#

收到，谢谢，查看一下^_^

赞(0）回复(0）举报 2021-12-07

tquggr8v4#

看到这个pr已经合并了，在as上使用! python -m pip install paddlepaddle-gpu==0.0.0.post101 -f https://paddlepaddle.org.cn/whl/cu101/mkl/develop.html装了最新的develop版本，但是使用还是有点问题。请问这个是我使用的问题还是不能支持这样使用。

IndexError: When index contains a Tensor, its length must be 1, but received 3

具体的代码中如下

F0[:, y0, x0]

其中

F0.shape: [64, 640, 480]
x0.shape: [14181] x0: Tensor(shape=[14181], dtype=int16, place=CUDAPlace(0), stop_gradient=True,
       [2  , 2  , 2  , ..., 476, 476, 477])
y0.shape: [14181] y0: Tensor(shape=[14181], dtype=int16, place=CUDAPlace(0), stop_gradient=True,
       [428, 142, 68 , ..., 227, 110, 398])

赞(0）回复(0）举报 2021-12-07

jmp7cifd5#

hi，我验证了一下，这个case下确实是还有一些问题。我会和相关负责的同学反馈下，看排期支持下这个用法。

赞(0）回复(0）举报 2021-12-07

kx5bkwkv6#

收到，感谢，待支持后马上体验进行反馈。

赞(0）回复(0）举报 2021-12-07

gc0ot86w7#

根据issues创建了任务卡片

赞(0）回复(0）举报 2021-12-07

y3bcpkx18#

xy = paddle.stack([y1,x1],axis=-1)
F1 = paddle.transpose(F0,[1,2,0])
F2 = paddle.gather_nd(F1,xy)
F3 = paddle.transpose(F2,[1,0])
感觉可以这样写