code: def get_shapes(symbol, data_shape):
      arg_name = symbol.list_arguments()
      arg_shape, _, _ = symbol.infer_shape(data=data_shape)
      shapes = [
          s
          for n, s in zip(arg_name, arg_shape)
          if "weight" in n or "bias" in n
      ]
      return shapes
  
file_path: mxnet/tools/bandwidth/measure.py
imports: import os, sys
  import mxnet
  import logging
  import argparse
  import time
  import numpy
  from importlib import import_module
  from collections import namedtuple
  from functools import reduce
label_desc: Use image node_id 2 for calling a global function with example usage: get_shapes(symbol, data_shape) and returns: shapes