flambe.nn.distance.hyperbolic
¶
Module Contents¶
-
flambe.nn.distance.hyperbolic.
project
(x)[source]¶ Project onto the hyeprboloid embedded in in n+1 dimensions.
-
class
flambe.nn.distance.hyperbolic.
HyperbolicDistance
[source]¶ Bases:
flambe.nn.distance.distance.DistanceModule
Implement a HyperbolicDistance 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.hyperbolic.
HyperbolicMean
[source]¶ Bases:
flambe.nn.distance.distance.MeanModule
Compute the mean point in the hyperboloid model.