code: def get_node_padding_mask(
      padded_graph: gn_graph.GraphsTuple,
  ) -> ArrayTree:
      """Returns a mask for the nodes of a padded graph.
  
      Args:
        padded_graph: ``GraphsTuple`` padded using ``pad_with_graphs``. This graph
          must contain at least one array of node features so the total static
          number of nodes can be inferred statically from the shape, and the method
          can be jitted.
  
      Returns:
        Boolean array of shape [total_num_nodes] containing True for real nodes,
        and False for padding nodes.
      """
      n_padding_node = get_number_of_padding_with_graphs_nodes(
          padded_graph
      )
      flat_node_features = tree.tree_leaves(padded_graph.nodes)
  
      if not flat_node_features:
          raise ValueError(
              "`padded_graph` must have at least one array of node features"
          )
      total_num_nodes = flat_node_features[0].shape[0]
      return _get_mask(
          padding_length=n_padding_node, full_length=total_num_nodes
      )
  
imports: import functools
  from typing import Any, Callable, Generator, Iterable, Iterator, List, Mapping, Optional, Sequence, Union
  import jax
  from jax import lax
  import jax.numpy
  import jax.tree_util
  from jraph._src import graph
  import numpy
label_desc: Use image node_id 23 for calling a global function with example usage: get_node_padding_mask(padded_graph) and returns: _get_mask