class_docstr: Compute the augmentation loss, as described in the
  `RangeAugment <https://arxiv.org/abs/2212.10553>`_ paper.
  
  Args:
      opts: command line arguments
class_docstr_tok: ["Compute","the","augmentation","loss",",","as","described","in","the","`","RangeAugment","<","https",":","\/\/arxiv.org\/abs\/2212.10553",">","`","_","paper",".","Args",":","opts",":","command","line","arguments"]
docstr: Compute the neural augmentation loss.
  
  Args:
      input_tensor: Input image of shape [N, C, H, W]
      augmented_tensor: Augmented image of shape [N, C, H, W]
  
  Returns:
      A scalar value
docstr_tok: ["Compute","the","neural","augmentation","loss",".","Args",":","input_tensor",":","Input","image","of","shape","[","N",",","C",",","H",",","W","]","augmented_tensor",":","Augmented","image","of","shape","[","N",",","C",",","H",",","W","]","Returns",":","A","scalar","value"]
code: def _compute_loss(
      self,
      input_tensor: Tensor,
      augmented_tensor: Tensor,
      *args,
      **kwargs,
  ) -> Tensor:
      """Compute the neural augmentation loss.
  
      Args:
          input_tensor: Input image of shape [N, C, H, W]
          augmented_tensor: Augmented image of shape [N, C, H, W]
  
      Returns:
          A scalar value
      """
      if augmented_tensor is None:
          logger.error(
              f"Augmented tensor can't be None in {self.__class__.__name__} during training mode."
          )
  
      forward_loss_fn = getattr(
          self, f"_forward_{self.perceptual_metric}"
      )
      loss_na = forward_loss_fn(
          input_tensor=input_tensor,
          augmented_tensor=augmented_tensor,
          *args,
          **kwargs,
      )
      return loss_na
  
file_path: ml-cvnets/loss_fn/neural_augmentation.py
imports: import argparse
  import math
  from typing import List, Mapping, Union
  import torch
  from torch import Tensor
  from torch.nn import functional
  from loss_fn import LOSS_REGISTRY, BaseCriteria
  from utils import logger
  from utils.ddp_utils import is_master
label_desc: Use image node_id 4 for calling the NeuralAugmentation obj's underlying member method code with example usage: obj._compute_loss(input_tensor, augmented_tensor) and returns: loss_na