index_sample¶
- paddle. index_sample ( x, index ) ¶
对输入 x
中的元素进行批量抽样,取 index
指定的对应下标的元素,按 index 中出现的先后顺序组织,填充为一个新的 Tensor。
x
与index
都是2-D
Tensor。index
的第一维度与输入x
的第一维度必须相同,index
的第二维度没有大小要求,可以重复索引相同下标元素。
参数¶
x (Tensor)– 输入的二维 Tensor,数据类型为 int32、int64、bfloat16、float16、float32、float64。
index (Tensor)– 包含索引下标的二维 Tensor。数据类型为 int32、int64。
返回¶
Tensor,数据类型与输入 x
相同,维度与 index
相同。
代码示例¶
COPY-FROM: paddle.index_sample