dgs.models.metric.metric.CosineDistanceMetric.forward

CosineDistanceMetric.forward(input1: torch.Tensor, input2: torch.Tensor) → torch.Tensor[source]
Parameters:

  • input1 – tensor of shape [a x E]

  • input2 – tensor of shape [b x E]

Returns:

tensor of shape [a x b] containing the distances.