dgs.models.metric.metric.custom_cosine_similarity

dgs.models.metric.metric.custom_cosine_similarity(input1: torch.Tensor, input2: torch.Tensor, dim: int, eps: float) torch.Tensor[source]

See https://github.com/pytorch/pytorch/issues/104564#issuecomment-1625348908