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

torch.nn.CosineSimilarity

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

paddle.nn.CosineSimilarity

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

两者功能一致且参数用法一致,仅参数名不同,具体如下:

参数映射

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