[ 仅参数名不一致 ]torch.nn.functional.cosine_similarity

torch.nn.functional.cosine_similarity

torch.nn.functional.cosine_similarity(x1,
                                      x2,
                                      dim=1,
                                      eps=1e-08,
                                      *,
                                      axis)

paddle.nn.functional.cosine_similarity

paddle.nn.functional.cosine_similarity(x1,
                                       x2,
                                       axis=1,
                                       eps=1e-8)

两者功能一致，仅参数名不一致，具体如下：

参数映射

PyTorch PaddlePaddle 备注
x1 x1 表示第一个输入的 Tensor 。
x2 x2 表示第二个输入的 Tensor 。
dim axis 表示计算的维度，仅参数名不一致。
eps eps 表示加到分母上的超参数 。
axis axis 表示计算的维度，dim 参数的替代。