code: def onnx_get_dynamic_axes(input_keys: List[str]):
      dynamic_axes = {}
      for k in input_keys:
          if "token_ids" in k or "segment_ids" in k:
              dynamic_axes[k] = {0: "batch_size", 1: "seq_length"}
          elif (
              "valid_length" in k
              or k.startswith("numerical")
              or k.startswith("timm_image")
          ):
              dynamic_axes[k] = {0: "batch_size"}
  
      return dynamic_axes
  
file_path: autogluon/multimodal/src/autogluon/multimodal/utils/onnx.py
imports: import logging
  import os
  from typing import Dict, List, Optional, Tuple, Union
  from torch import tensor
  from ..constants import AUTOMM, FEATURE_EXTRACTION, MULTICLASS
label_desc: Use image node_id 1 for calling a global function with example usage: onnx_get_dynamic_axes(input_keys) and returns: dynamic_axes