CosineSimilarity

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

CosineSimilarity

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

参数:

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

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

返回:无

代码示例

  1. import paddle
  2. import paddle.nn as nn
  3. import numpy as np
  4. np.random.seed(0)
  5. x1 = np.random.rand(2,3)
  6. x2 = np.random.rand(2,3)
  7. x1 = paddle.to_tensor(x1)
  8. x2 = paddle.to_tensor(x2)
  9. cos_sim_func = nn.CosineSimilarity(axis=0)
  10. result = cos_sim_func(x1, x2)
  11. print(result)
  12. # [0.99806249 0.9817672 0.94987036]