2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.IndexPut.rst 2.06 KB
一键复制 编辑 原始数据 按行查看 历史
luojianing 提交于 2023-06-25 20:32 . fix docs issues

mindspore.ops.IndexPut

.. py:class:: mindspore.ops.IndexPut(accumulate=0)

    根据 `indices` 中的下标值,使用 `x2` 中的数值替换 `x1` 中的相应元素的值。

    参数:
        - **accumulate** (int) - 如果 `accumulate` 被设置为1 , `x2` 中的元素被累加到 `x1` 的相应元素上;
          如果为 ``0``,用 `x2` 中的元素取代 `x2` 的相应元素。默认值: ``0`` 。

    输入:
        - **x1** (Tensor) - 被执行替换操作的Tensor, 维度大于等于1。
        - **x2** (Tensor) - 数据类型和 `x1` 一致的一维的Tensor。如果其size为1,则shape将被广播为 `x1` 的shape。
        - **indices** (tuple[Tensor], list[Tensor]) - 元素类型是int32或者int64, 用于对 `x1` 中的元素进行索引。
          `indices` 中的tensor的秩应为1-D, `indices` 中tensor的size应小于 `x1` 的秩,indices中的tensor应是可广播的。

    输出:
        Tensor, 其数据类型和shape与  `x1` 相同。

    异常:
        - **TypeError** - 如果 `x1` 的dtype与 `x2` 的dtype不同。
        - **TypeError** - 如果 `indices` 不是tuple[Tensor]或者list[Tensor]。
        - **TypeError** - 如果 `indices` 中的tensor的dtype不是int32或者int64。
        - **TypeError** - 如果 `indices` 中的tensor的dtype是不一致的。
        - **TypeError** - 如果 `accumulate` 的dtype不是int。
        - **ValueError** - 如果 `x2` 的秩不是1-D。
        - **ValueError** - 当 `x1` 的rank与 `indices` 的size相等时,如果 `x2` 的size不为1
          或者不为 `indices` 中Tensor的最大size。
        - **ValueError** - 当 `x1` 的rank大于 `indices` 的size时,如果 `x2` 的size不为1
          或者不为 `x1` 的最后一维的shape。
        - **ValueError** - 如果 `indices` 中的tensor的秩不是1-D。
        - **ValueError** - 如果 `indices` 中的tensor不是可广播的。
        - **ValueError** - 如果 `indices` 的size大于 `x1` 的秩。
        - **ValueError** - 如果 `accumulate` 的值不是0或1。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.1

搜索帮助