cosine_similarity

    该OP用于计算x1与x2沿axis维度的余弦相似度。

    参数:

    返回: 余弦相似度的计算结果,数据类型与x1, x2相同。

    返回类型:Tensor