How do you ensure sensible ranges for input to MixtureNormal layers?

Question

I usually see the tfp MixtureNormal layer used as follows:

params_size = tfp.layers.MixtureNormal.params_size(
    num_