code: def get_node_shape(node: Node) -> Any:
      """
      return node data shape
      """
      if get_node_name(node) in ["split", "unbind"]:
          return node.meta["tensor_meta"][0].shape
      if hasattr(node.meta["tensor_meta"], "shape"):
          return node.meta["tensor_meta"].shape
      return None
  
imports: from typing import Any, Dict, List, Union
  from torch.fx.node import Node
  from colossalai.logging import get_dist_logger
label_desc: Use image node_id 5 for calling a global function with example usage: get_node_shape(node) and returns: None, node, node