2.3K Star 8.1K Fork 4.3K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_cdist.rst 1.38 KB
一键复制 编辑 原始数据 按行查看 历史

mindspore.ops.cdist

.. py:function:: mindspore.ops.cdist(x1, x2, p=2.0)

    计算两个Tensor每对行向量之间的p-norm距离。

    .. note::
        Ascend上支持的输入数据类型为[float16, float32],CPU上支持的输入数据类型为[float16, float32],GPU上支持的输入数据类型为[float32, float64]。

    参数:
        - **x1** (Tensor) - 输入Tensor,shape为 :math:`(B, P, M)` , :math:`B` 表示0或者正整数。 :math:`B` 维度为0时该维度被忽略,shape为 :math:`(P, M)` 。
        - **x2** (Tensor) - 输入Tensor,shape为 :math:`(B, R, M)` ,与 `x1` 的数据类型一致。
        - **p** (float,可选) - 计算向量对p-norm距离的P值,P∈[0,∞]。默认值: ``2.0`` 。

    返回:
        Tensor,p-范数距离,数据类型与 `x1` 一致,shape为 :math:`(B, P, R)`。

    异常:
        - **TypeError** - `x1` 或 `x2` 不是Tensor。
        - **TypeError** - `x1` 或 `x2` 的数据类型不符合上述“说明”中的要求。
        - **TypeError** - `p` 不是float32。
        - **ValueError** - `p` 是负数。
        - **ValueError** - `x1` 与 `x2` 维度不同。
        - **ValueError** - `x1` 与 `x2` 的维度既不是2,也不是3。
        - **ValueError** - 单批次训练下 `x1` 和 `x2` 的shape不一样。
        - **ValueError** - `x1` 和 `x2` 的列数不一样。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助