tfp.substrates.jax.util.BatchedComponentProperties

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

Inherits From: ParameterProperties

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

Computes parameter event_ndims when parameterizing a bijector.

instance_event_ndims

View source

NO_EVENT_NDIMS 'INTERNAL_NO_EVENT_NDIMS'