API referenceΒΆ
One page per part of the library, in dependency order: models are built from layers, layers from functions. Each page starts with what the objects are for and a summary table; the detailed entries follow. For the reasoning behind them (geometries, gradient paths, parametrizations), see the User guide.
Page |
Contents |
|---|---|
|
|
|
|
Riemannian batch normalization (mean only, mean + scalar variance, GBWBN) |
|
Spectral vector field and residual block of the residual networks |
|
SPD, Stiefel and positive-scalar parametrizations of the parameters |
|
Matrix functions through eigendecomposition, congruences, vectorizations |
|
Geodesics, means, dispersions, exp/log maps of the five geometries |
|
Sample covariance and robust M-estimators of scatter |
|
Scalar maps, Stiefel projections, random SPD/Stiefel matrices |