2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.numpy.indices.rst 1.01 KB
一键复制 编辑 原始数据 按行查看 历史
俞涵 提交于 2024-09-13 15:59 . modify numpy docs format errors

mindspore.numpy.indices

.. py:function:: mindspore.numpy.indices(dimensions, dtype=mstype.int32, sparse=False)

    返回一个表示网格索引的数组。
    计算一个数组,其中子数组包含仅沿相应轴变化的索引值0,1,...。

    参数:
        - **dimensions** (Union[list(int), tuple]) - 网格的shape。
        - **dtype** (mindspore.dtype, 可选) - 指定结果的数据类型。
        - **sparse** (boolean, 可选) - 返回网格的稀疏表示而非密集表示,默认值: ``False`` 。

    返回:
        Tensor,或元素为Tensor的Tuple。如果 ``sparse`` 为 ``False`` ,则返回一个网格索引数组: ``grid.shape = (len(dimensions),) + tuple(dimensions)`` 。如果 ``sparse`` 为 ``True`` ,则返回数组的Tuple,其中 ``grid[i].shape = (1, ..., 1, dimensions[i], 1, ..., 1)`` , ``dimensions[i]`` 位于第 ``i`` 个位置。

    异常:
        - **TypeError** - 如果输入的 ``dimensions`` 不是Tuple或List。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助