class_docstr: A Zero Inflated Negative Binomial distribution.
  
  :param total_count: non-negative number of negative Bernoulli trials.
  :type total_count: float or torch.Tensor
  :param torch.Tensor probs: Event probabilities of success in the half open interval [0, 1).
  :param torch.Tensor logits: Event log-odds for probabilities of success.
  :param torch.Tensor gate: probability of extra zeros.
  :param torch.Tensor gate_logits: logits of extra zeros.
class_docstr_tok: ["A","Zero","Inflated","Negative","Binomial","distribution",".",":","param","total_count",":","non-negative","number","of","negative","Bernoulli","trials",".",":","type","total_count",":","float","or","torch.Tensor",":","param","torch.Tensor","probs",":","Event","probabilities","of","success","in","the","half","open","interval","[","0",",","1",")",".",":","param","torch.Tensor","logits",":","Event","log-odds","for","probabilities","of","success",".",":","param","torch.Tensor","gate",":","probability","of","extra","zeros",".",":","param","torch.Tensor","gate_logits",":","logits","of","extra","zeros","."]
code: def total_count(self):
      return self.base_dist.total_count
  
imports: import torch
  from torch.distributions import constraints
  from torch.distributions.utils import broadcast_all, lazy_property, logits_to_probs, probs_to_logits
  from torch.nn.functional import softplus
  from pyro.distributions import NegativeBinomial, Poisson, TorchDistribution
  from pyro.distributions.util import broadcast_shape
label_desc: Use image node_id 2 for calling the ZeroInflatedNegativeBinomial obj's underlying member method code with example usage: obj.total_count() and returns: self