ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfp.experimental.tangent_spaces

TensorFlow Probability experimental tangent spaces package.


class AxisAlignedSpace: Tangent space of M for a distribution with axis-aligned subspace support.

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

class GeneralSpace: Arbitrary tangent space when no more-efficient special case applies.

class TangentSpace: Represents a tangent space to some manifold M at a point x.

class ZeroSpace: Tangent space of M for discrete distributions.