code: def forward(self, *input, **kwargs):
      output = self.model(*input, **kwargs)
      for key in self.output_keys:
          output[key] = (
              output[key].mul(self.temperature).softmax(dim=self.dim)
          )
      return output
  
imports: import torch
  from torch import nn, Tensor
  from typing import List, Union, Iterable, Optional, Dict, Tuple
  from pytorch_toolbelt.inference.tta import _deaugment_averaging
label_desc: Use image node_id 2 for calling the ApplySoftmaxTo obj's underlying member method code with example usage: obj.forward() and returns: output