gather_nd

• gather_nd(input, index, name=None)[源代码]

示例：

• 参数：

返回类型：Variable

x = fluid.layers.data(name='x', shape=[3, 4, 5], dtype='float32')
index = fluid.layers.data(name='index', shape=[2, 2], dtype='int32')
output = fluid.layers.gather_nd(x, index)