gather

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

    参数:

    • index (Tensor) - 索引 Tensor，秩 rank = 1, 数据类型为 int32 或 int64。

    • axis (Tensor) - 指定index 获取输入的维度， axis 的类型可以是int或者Tensor，当 为Tensor的时候其数据类型为int32 或者int64。

    返回：和输入的秩相同的输出Tensor。

    代码示例：

    1. import numpy as np
    2. import paddle
    5. input = paddle.to_tensor(input_1)
    6. index = paddle.to_tensor(index_1)
    7. # expected output: [[1,2],[3,4]]