首页 Paddle框架 帖子详情
variable如何按索引获取值?
收藏
快速回复
Paddle框架 问答深度学习 886 1
variable如何按索引获取值?
收藏
快速回复
Paddle框架 问答深度学习 886 1

假设有一个mini batch, 用variable变量a表示。有一个列表b存放索引, 目的是利用b中的索引获取变量a中的值。

如下图所示一个简化例子。之前看过别人在pytorch上直接用a[:,b],但在paddle里这样尝试却报错了,一直尝试没找到相应的方法,求解。

顺便附上简化代码

import paddle
import paddle.fluid as fluid

with fluid.dygraph.guard():
    a = np.array([[1,3,5,8],[0,6,1,9],[2,5,2,1],[6,5,4,4]])
    a = fluid.dygraph.to_variable(a)
    #print("a =", a)

    b = [4,3,3,1]
    #print("b =", b)

    result = a[:,b]
    print(result)
0
收藏
回复
全部评论(1)
时间顺序
m
mar836924042
#2 回复于2020-08

b写错了, 应该是[3,2,2,0]

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