tensorflow_probability/python/layers/distribution_layer.py
def params_size(event_size, num_components, name=None):
    """The number of `params` needed to create a single distribution."""
    with tf.compat.v1.name_scope(
        name, 'CategoricalMixtureOfOneHotCategorical_params_size',
        [event_size, num_components]):
      return MixtureSameFamily.params_size(
          num_components,
          OneHotCategorical.params_size(event_size, name=name),
          name=name)
def params_size(event_size, num_components, name=None):
    """The number of `params` needed to create a single distribution."""
    with tf.compat.v1.name_scope(
        name, 'CategoricalMixtureOfOneHotCategorical_params_size',
        [event_size, num_components]):
      return MixtureSameFamily.params_size(
          num_components,
          OneHotCategorical.params_size(event_size, name=name),
          name=name)
def params_size ( event_size , num_components , name = None ) : with tf . compat . v1 . name_scope ( name , 'CategoricalMixtureOfOneHotCategorical_params_size' , [ event_size , num_components ] ) : return MixtureSameFamily . params_size ( num_components , OneHotCategorical . params_size ( event_size , name = name ) , name = name )
The number of `params` needed to create a single distribution.
The number of params needed to create a single distribution .