index_sample

    该OP实现对输入 x 中的元素进行批量抽样,取 index 指定的对应下标的元素,按index中出现的先后顺序组织,填充为一个新的张量。

    该OP中 与 index 都是 2-D 张量。 的第一维度与输入 x 的第一维度必须相同, index 的第二维度没有大小要求,可以重复索引相同下标元素。

    -Tensor ,数据类型与输入 相同,维度与 index 相同。

    代码示例