code: def torch_bmm(
      input: torch.Tensor,
      other: torch.Tensor,
      *,
      out: Optional[torch.Tensor] = None,
  ) -> Tuple[int, int]:
      macs = reduce(operator.mul, input.shape) * other.shape[-1]
      flops = 2 * macs
      return flops, macs
  
file_path: ColossalAI/colossalai/fx/profiler/experimental/profiler_function/arithmetic.py
imports: import operator
  from functools import reduce
  from typing import Any, Optional, Tuple, Union
  import torch
  from ..registry import meta_profiler_function
label_desc: Use image node_id 5 for calling a global function with example usage: torch_bmm(input, other) and returns: flops,  macs