2.3K Star 8.1K Fork 4.3K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_median.rst 1.42 KB
一键复制 编辑 原始数据 按行查看 历史
wind-zyx 提交于 2023-04-20 20:21 . ops update value format 0420

mindspore.ops.median

.. py:function:: mindspore.ops.median(input, axis=-1, keepdims=False)

    输出Tensor指定维度上的中值与索引。

    .. warning::
        - 如果 `input` 的中值不唯一,则 `indices` 不一定包含第一个出现的中值。该接口的具体实现方式和后端类型相关,CPU和GPU的返回值可能不相同。

    参数:
        - **input** (Tensor) - median的输入,任意维度的Tensor。数据类型支持int16、int32、int64、float32或float64。
        - **axis** (int,可选) - 指定计算维度。默认值: ``-1`` 。取值范围为[-dims, dims - 1],`dims` 表示 `input` 的维度长度。
        - **keepdims** (bool,可选) - 表示是否减少维度,如果为 ``True`` ,输出将与输入保持相同的维度;如果为 ``False`` ,输出将减少维度。默认值: ``False`` 。

    返回:
        - **y** (Tensor) - 返回指定维度上的中值,数据类型与 `input` 相同。
        - **indices** (bool) - 指定中值索引。数据类型为int64。

    异常:
        - **TypeError** - `input` 的数据类型不是: int16、int32、int64、float32、float64。
        - **TypeError** - `input` 不是Tensor。
        - **TypeError** - `axis` 不是int。
        - **TypeError** - `keepdims` 不是bool。
        - **ValueError** - `axis` 的范围不在[-dims, dims - 1],`dims` 表示 `x` 的维度长度。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助