tfp.experimental.distributions.marginal_fns.ps.size0

Returns the size of the first dimension (0 if scalar).