mgplvm.manifolds.so3 module

class mgplvm.manifolds.so3.So3(m, d=None)[source]

Bases: mgplvm.manifolds.base.Manifold

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

x, y: (…, n x d x m)

Return type

Tensor

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

x in R^3 -> q in R^4 s.t. ||q|| = 1

Return type

Tensor

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

Tensor

static gmul(x, y)[source]
Return type

Tensor

inducing_points(n, n_z, z=None)[source]
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]

phi = |x|/2

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

q in R^4 s.t. ||q|| = 1 -> x in R^3

Return type

Tensor

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

Tensor

parameterise_inducing(x)[source]