gather
根据索引 index 获取输入 x
的指定 aixs
维度的条目,并将它们拼接在一起。
参数:
index (Tensor) - 索引 Tensor,秩
rank = 1
, 数据类型为 int32 或 int64。axis (Tensor) - 指定index 获取输入的维度, 的类型可以是int或者Tensor,当
axis
为Tensor的时候其数据类型为int32 或者int64。
返回:和输入的秩相同的输出Tensor。
代码示例:
import numpy as np
import paddle
index_1 = np.array([0,1])
input = paddle.to_tensor(input_1)
index = paddle.to_tensor(index_1)