flambe.nn.distance.euclidean
¶
Module Contents¶
-
class
flambe.nn.distance.euclidean.
EuclideanDistance
[source]¶ Bases:
flambe.nn.distance.distance.DistanceModule
Implement a EuclideanDistance object.
-
forward
(self, mat_1: Tensor, mat_2: Tensor)[source]¶ Returns the squared euclidean distance between each element in mat_1 and each element in mat_2.
Parameters: - mat_1 (torch.Tensor) – matrix of shape (n_1, n_features)
- mat_2 (torch.Tensor) – matrix of shape (n_2, n_features)
Returns: dist – distance matrix of shape (n_1, n_2)
Return type: torch.Tensor
-
-
class
flambe.nn.distance.euclidean.
EuclideanMean
[source]¶ Bases:
flambe.nn.distance.distance.MeanModule
Implement a EuclideanMean object.