code: def domain_map(  # type: ignore
      cls,
      loc: torch.Tensor,
      scale: torch.Tensor,
  ):
      scale = F.softplus(scale)
  
      return (
          loc.squeeze(-1),
          scale.squeeze(-1),
      )
  
imports: import math
  from numbers import Number
  from typing import Dict, Optional, Tuple, Union
  import torch
  from torch.distributions import constraints
  from torch.distributions.utils import broadcast_all
  import torch.nn.functional
  from .distribution_output import DistributionOutput
  from gluonts.core.component import validated
  from torch.distributions import Distribution
label_desc: Use image node_id 2 for calling the TruncatedNormalOutput obj's underlying member method code with example usage: obj.domain_map(cls, loc, scale) and returns: loc,  scale