code: def domain_map(cls, total_count: torch.Tensor, logits: torch.Tensor):  # type: ignore
      total_count = F.softplus(total_count)
      return total_count.squeeze(-1), logits.squeeze(-1)
  
file_path: gluonts/src/gluonts/torch/distributions/negative_binomial.py
imports: from typing import Dict, Optional, Tuple, Union
  import torch
  import torch.nn.functional
  from torch.distributions import Distribution
  from torch.distributions import NegativeBinomial
  from gluonts.util import lazy_property
  from scipy.stats import nbinom
  from .distribution_output import DistributionOutput
label_desc: Use image node_id 1 for calling the NegativeBinomialOutput obj's underlying member method code with example usage: obj.domain_map(cls, total_count, logits) and returns: total_count,  logits