cleanUrl: "pytorch-cosine-similarity"
description: "PyTorch에서 cosine similarity 레이어를 만드는 법을 정리합니다."

Quickstart

import torch.nn as nn

v1 = torch.randn(100, 128)
v2 = torch.randn(100, 128)

cos = nn.CosineSimilarity(dim=1, eps=1e-6)
# dim = 어느 dimension으로 cosine similarity를 구할 지 정해줌.
out = cos(v1, v2)