Fix gate & gate_logits in ZeroInflatedDistribution
#3303
Merged
gate & gate_logits in ZeroInflatedDistribution
#3303