mindspore.train.CosineSimilarity

.. py:class:: mindspore.train.CosineSimilarity(similarity='cosine', reduction='none', zero_diagonal=True)

    计算余弦相似度。

    参数：
        - **similarity** (str) - 计算逻辑。 ``"cosine"`` 表示相似度计算逻辑， ``"dot"`` 表示矩阵点乘计算逻辑。默认值： ``'cosine'`` 。
        - **reduction** (str) - 规约计算方式。支持 ``"none"`` 、 ``"sum"`` 或 ``"mean"`` 。默认值： ``'none'`` 。
        - **zero_diagonal** (bool) - 如果为 ``True`` ，则对角线将设置为零。默认值： ``True`` 。

    .. py:method:: clear()

        重置评估结果。

    .. py:method:: eval()

        计算Cosine Similarity矩阵。

        返回：
            numpy.ndarray，相似度矩阵。

        异常：
            - **RuntimeError** - 如果没有先调用update方法。

    .. py:method:: update(*inputs)

        使用 `inputs` 更新内部评估结果。

        参数：
            - **inputs** (Union[Tensor, list, numpy.ndarray]) - 输入的矩阵，包含 `y_pred` 和 `y` 。