Skip to content

Latest commit

 

History

History
42 lines (27 loc) · 1.14 KB

gather_cn.rst

File metadata and controls

42 lines (27 loc) · 1.14 KB
Raw

gather

根据索引 index 获取输入 x 的指定 aixs 维度的条目，并将它们拼接在一起。

Given:

X = [[1, 2],
     [3, 4],
     [5, 6]]

Index = [1, 2]

axis = 0

Then:

Out = [[3, 4],
       [5, 6]]

参数

  • x (Tensor) - 输入 Tensor，秩 rank >= 1，支持的数据类型包括 int32、int64、float32、float64 和 uint8 (CPU)、float16（GPU） 。
  • index (Tensor) - 索引 Tensor，秩 rank = 1，数据类型为 int32 或 int64。
  • axis (Tensor) - 指定 index 获取输入的维度，axis 的类型可以是 int 或者 Tensor，当 axis 为 Tensor 的时候其数据类型为 int32 或者 int64。
  • name (str，可选) - 具体用法请参见 api_guide_Name，一般无需设置，默认值为 None。

返回

和输入的秩相同的输出 Tensor。

代码示例

COPY-FROM: paddle.gather