CosineSimilarity¶

class paddle.nn. CosineSimilarity ( axis=1, eps=1e-8 ) [源代码] ¶

CosineSimilarity

计算x1与x2沿axis维度的余弦相似度。

参数：

• axis (int) - 指定计算的维度，会在该维度上计算余弦相似度，默认值为1。

• eps (float) - 很小的值，防止计算时分母为0，默认值为1e-8。

返回：无

代码示例

import paddle
import paddle.nn as nn
import numpy as np

np.random.seed(0)
x1 = np.random.rand(2,3)
x2 = np.random.rand(2,3)
x1 = paddle.to_tensor(x1)
x2 = paddle.to_tensor(x2)

cos_sim_func = nn.CosineSimilarity(axis=0)
result = cos_sim_func(x1, x2)
print(result)
# [0.99806249 0.9817672  0.94987036]