[ 仅参数名不一致 ]torch.nn.CosineSimilarity¶

torch.nn.CosineSimilarity ¶

          torch.nn.CosineSimilarity(dim=1,
                            eps=1e-08)

          paddle.nn.CosineSimilarity(axis=1,
                            eps=1e-8)

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

PyTorch	PaddlePaddle	备注
dim	axis	指定计算的维度，会在该维度上计算余弦相似度，默认值为 1。
eps	eps	表示计算余弦相似度公式中的超参数。