tfp.substrates.jax.util.BatchedComponentProperties

Alias to assist in defining properties of non-Tensor parameters.

Inherits From: ParameterProperties

View aliases

Main aliases

tfp.experimental.substrates.jax.util.BatchedComponentProperties

tfp.substrates.jax.util.BatchedComponentProperties(
    event_ndims=0,
    event_ndims_tensor=None,
    default_constraining_bijector_fn=None,
    is_preferred=True
)

Attributes
event_ndims	A namedtuple alias for field number 0
event_ndims_tensor	A namedtuple alias for field number 1
shape_fn	A namedtuple alias for field number 2
default_constraining_bijector_fn	A namedtuple alias for field number 3
is_preferred	A namedtuple alias for field number 4
is_tensor	A namedtuple alias for field number 5
specifies_shape	A namedtuple alias for field number 6

Methods

bijector_instance_event_ndims

View source

bijector_instance_event_ndims(
    bijector, x_event_ndims, require_static=False
)

Computes parameter event_ndims when parameterizing a bijector.

instance_event_ndims

View source

instance_event_ndims(
    instance, require_static=False
)

Class Variables
NO_EVENT_NDIMS	'INTERNAL_NO_EVENT_NDIMS'