code: def _shape(tf_expr, dim_idx):
      if tf_expr.shape.rank is not None:
          dim = tf_expr.shape[dim_idx].value
          if dim is not None:
              return dim
      return tf.shape(tf_expr)[dim_idx]
  
file_path: insightface/reconstruction/ostec/external/stylegan2/dnnlib/tflib/ops/upfirdn_2d.py
imports: import os
  import numpy
  import tensorflow
  from ..None import custom_ops
label_desc: Use image node_id 10 for calling a global function with example usage: _shape(tf_expr, dim_idx) and returns: tf, dim