code: def _get_graph_size(graphs_tuple):
      n_node = np.sum(graphs_tuple.n_node)
      n_edge = len(graphs_tuple.senders)
      n_graph = len(graphs_tuple.n_node)
      return n_node, n_edge, n_graph
  
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 31 for calling a global function with example usage: _get_graph_size(graphs_tuple) and returns: n_node,  n_edge,  n_graph