tfp.experimental.tangent_spaces.FullSpace

Tangent space of M for distributions supported on all of R^n.

Inherits From: TangentSpace

This subclass covers full-rank distributions on n-dimensional manifolds. In this common case we can take the basis to be the standard basis for R^n, so we need not explicitly represent it at all.

Methods

`transform_coordinatewise`

View source

transform_coordinatewise(
    x, f, **kwargs
)

Same as transform_dimension_preserving, for a coordinatewise f.

Default falls back to transform_dimension_preserving, which may be overridden in subclasses.

Args
`x`	same as in `transform_dimension_preserving`.
`f`	same as in `transform_dimension_preserving`.
`**kwargs`	same as in `transform_dimension_preserving`.

Returns
`log_density`	A `Tensor` representing the log density correction of f at x
`space`	A `TangentSpace` representing the tangent to fM at f(x)

Raises
`NotImplementedError`	if the `TangentSpace` subclass does not implement `transform_dimension_preserving`.

`transform_dimension_preserving`

View source

transform_dimension_preserving(
    x, f, **kwargs
)

Same as transform_general, assuming f goes from R^n to R^n.

Default falls back to transform_general, which may be overridden in subclasses.

Args
`x`	same as in `transform_general`.
`f`	same as in `transform_general`.
`**kwargs`	same as in `transform_general`.

Returns
`log_density`	A `Tensor` representing the log density correction of f at x
`space`	A `TangentSpace` representing the tangent to fM at f(x)

Raises
`NotImplementedError`	if the `TangentSpace` subclass does not implement `transform_general`.

`transform_general`

View source

transform_general(
    x, f, **kwargs
)

If the bijector is weird, fall back to the general case.

`transform_projection`

View source

transform_projection(
    x, f, **kwargs
)

Same as transform_general, with f a projection (or its inverse).

Default falls back to transform_general, which may be overridden in subclasses.

Args
`x`	same as in `transform_general`.
`f`	same as in `transform_general`.
`**kwargs`	same as in `transform_general`.

Returns
`log_density`	A `Tensor` representing the log density correction of f at x
`space`	A `TangentSpace` representing the tangent to fM at f(x)

Raises
`NotImplementedError`	if the `TangentSpace` subclass does not implement `transform_general`.