code: def mask_leading_dimension(tensor):
      # Place holder masking does not work anymore. Just use the tensor as is.
      return tf.constant(tensor, shape=tensor.get_shape().as_list())
  
file_path: graph_nets/graph_nets/tests_tf2/test_utils.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import contextlib
  import itertools
  from graph_nets import graphs
  from graph_nets import utils_np
  import numpy
  import tensorflow
label_desc: Use image node_id 2 for calling a global function with example usage: mask_leading_dimension(tensor) and returns: tf