mgplvm.manifolds.s3 module

class mgplvm.manifolds.s3.S3(m, d=None)[source]

Bases: mgplvm.manifolds.base.Manifold

static distance(x, y, ell=None)[source]
Return type

Tensor

static expmap(x, dim=- 1)[source]

same as SO(3)

Return type

Tensor

static expmap2(x, dim=- 1)[source]
Return type

Tensor

static gmul(x, y)[source]

same as SO(3)

Return type

Tensor

inducing_points(n, n_z, z=None)[source]
static initialize(initialization, n_samples, m, d, Y)[source]

initializes latents - can add more exciting initializations as well

static inverse(q)[source]
Return type

Tensor

static log_q(log_base_prob, x, d, kmax, dim=- 1)[source]

theta = |x|/2

log_uniform = -2.9826069522587457
static logmap(q, dim=- 1)[source]

same as SO(3)

Return type

Tensor

lprior(g)[source]
property name
static parameterise(x)[source]
Return type

Tensor