variable如何按索引获取值?
收藏
假设有一个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
收藏
请登录后评论
b写错了, 应该是[3,2,2,0]