gather
paddle.fluid.layers.
gather
(input, index, overwrite=True)[源代码]
根据索引 index
获取输入(input)的最外层维度的条目,并将它们拼接在一起。
- X = [[1, 2],
- [3, 4],
- [5, 6]]
- Index = [1, 2]
- Then:
- Out = [[3, 4],
- [5, 6]]
- 参数:
- input (Variable) - 输入, 秩
rank >= 1
, 支持的数据类型包括 int32、int64、float32、float64 和 uint8 (CPU)、float16(GPU) 。 - index (Variable) - 索引,秩
rank = 1
, 数据类型为 int32 或 int64。 - overwrite (bool) - 具有相同索引时在反向更新梯度的模式。如果为
True
,则使用覆盖模式更新相同索引的梯度;如果为False
,则使用累积模式更新相同索引的梯度。默认值为True
。
- input (Variable) - 输入, 秩
返回:和输入的秩相同的输出张量。
返回类型:Variable
代码示例
- import paddle.fluid as fluid
- x = fluid.layers.data(name='x', shape=[-1, 5], dtype='float32')
- index = fluid.layers.data(name='index', shape=[-1, 1], dtype='int32')
- output = fluid.layers.gather(x, index)