class_docstr: Anomaly Map Generator for CS-Flow model.
  
  Args:
      input_dims (tuple[int, int, int]): Input dimensions.
      mode (AnomalyMapMode): Anomaly map mode. Defaults to AnomalyMapMode.ALL.
code: def __init__(
      self,
      input_dims: tuple[int, int, int],
      mode: AnomalyMapMode = AnomalyMapMode.ALL,
  ) -> None:
      super().__init__()
      self.mode = mode
      self.input_dims = input_dims
  
file_path: anomalib/src/anomalib/models/csflow/anomaly_map.py
imports: from __future__ import annotations
  from enum import Enum
  import torch
  import torch.nn.functional
  from torch import Tensor, nn
label_desc: Use image node_id 1 to create a new AnomalyMapGenerator object from inherited base classes: nn with example: obj = AnomalyMapGenerator(input_dims, mode)