Create a random variable for MixtureSameFamily.
tfp.edward2.MixtureSameFamily( *args, **kwargs )
See MixtureSameFamily for more details.
Original Docstring for Distribution
tfp.distributions.Categorical-like instance. Manages the probability of selecting components. The number of categories must match the rightmost batch dimension of the
components_distribution. Must have either scalar
tfp.distributions.Distribution-like instance. Right-most batch dimension indexes components.
False. Whether to reparameterize samples of the distribution using implicit reparameterization gradients [(Figurnov et al., 2018)]. The gradients for the mixture logits are equivalent to the ones described by [(Graves, 2016)]. The gradients for the components parameters are also computed using implicit reparameterization (as opposed to ancestral sampling), meaning that all components are updated every step. Only works when: (1) components_distribution is fully reparameterized; (2) components_distribution is either a scalar distribution or fully factorized (tfd.Independent applied to a scalar distribution); (3) batch shape has a known rank. Experimental, may be slow and produce infs/NaNs.
Truedistribution parameters are checked for validity despite possibly degrading runtime performance. When
Falseinvalid inputs may silently render incorrect outputs.
True, statistics (e.g., mean, mode, variance) use the value "
NaN" to indicate the result is undefined. When
False, an exception is raised if one or more of the statistic's batch members are undefined.
strname prefixed to Ops created by this class.
if not dtype_util.is_integer(mixture_distribution.dtype).
ValueError: if mixture_distribution does not have scalar
components_distribution.batch_shape[:-1]are both fully defined and the former is neither scalar nor equal to the latter.
mixture_distributioncategories does not equal
components_distributionrightmost batch shape.
: Michael Figurnov, Shakir Mohamed and Andriy Mnih. Implicit reparameterization gradients. In Neural Information Processing Systems, 2018. https://arxiv.org/abs/1805.08498
: Alex Graves. Stochastic Backpropagation through Mixture Density Distributions. arXiv, 2016. https://arxiv.org/abs/1607.05690