code: def clear_dims(infshape_dict):
      """
      Input:
          infshape_dict: dict of `InfShape`
      Output:
          the same dict but where all `InfDim` in all `InfShape`
          have their `dim` attribute set to None
      """
      d = deepcopy(infshape_dict)
      for _, v in d.items():
          for infdim in v:
              infdim.dim = None
      return d
  
file_path: NeMo/nemo/collections/nlp/modules/common/megatron/mup/shape.py
imports: from copy import deepcopy
  import yaml
  from torch import nn
  from torch.nn import Linear
  from torch.nn.modules.conv import _ConvNd
  from nemo.collections.nlp.modules.common.megatron.mup.infshape import InfShape, zip_infshape
  from nemo.collections.nlp.modules.common.megatron.mup.layer import MuReadout, rescale_linear_bias
label_desc: Use image node_id 9 for calling a global function with example usage: clear_dims(infshape_dict) and returns: d