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
)
Methods
bijector_instance_event_ndims
bijector_instance_event_ndims(
bijector, x_event_ndims, require_static=False
)
Computes parameter event_ndims when parameterizing a bijector.
instance_event_ndims
instance_event_ndims(
instance, require_static=False
)