code: def _get_reduction_function_output_dim(
      reduction_function: ReductionFunction,
      feat_dim: int,
  ) -> int:
      if reduction_function == ReductionFunction.STD_AVG:
          return 1
      else:
          return feat_dim
  
imports: from abc import ABC, abstractmethod
  from enum import Enum
  from typing import Dict, Optional, Sequence, Tuple, Union
  import torch
  import torch.nn.functional
  from pytorch3d.implicitron.models.view_pooler.view_sampler import cameras_points_cartesian_product
  from pytorch3d.implicitron.tools.config import registry, ReplaceableBase
  from pytorch3d.ops import wmean
  from pytorch3d.renderer.cameras import CamerasBase
label_desc: Use image node_id 2 for calling a global function with example usage: _get_reduction_function_output_dim(reduction_function, feat_dim) and returns: int, feat_dim