2.5K Star 8.4K Fork 4.7K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.SparseGatherV2.rst 640 Bytes
一键复制 编辑 原始数据 按行查看 历史
Li Qingguo 提交于 2023-06-30 09:32 . update doc

mindspore.ops.SparseGatherV2

.. py:class:: mindspore.ops.SparseGatherV2

    基于指定的索引和axis返回输入Tensor的切片。

    输入:
        - **input_params** (Tensor) - 被切片的Tensor。shape: :math:`(x_1, x_2, ..., x_R)` 。
        - **input_indices** (Tensor)- shape: :math:`(y_1, y_2, ..., y_S)` 。
          指定切片的索引,取值须在 `[0, input_params.shape[axis])` 范围内。
        - **axis** (Union(int, Tensor[int])) - 进行索引的axis。axis是Tensor的时候,size必须是1。

    输出:
        Tensor,shape: :math:`(z_1, z_2, ..., z_N)` 。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.q1

搜索帮助