code: def get_graphs_tuple_size(graphs_tuple):
      """Calculates the total nodes, edges and graphs in a graph batch.
  
      Args:
        graphs_tuple: A `GraphsTuple`.
  
      Returns:
        A `GraphsTupleSizes` object containing the total number of nodes, edges and
        graphs in the `GraphsTuple`. Each value is a scalar integer `tf.Tensor`.
  
      """
      num_nodes = tf.reduce_sum(graphs_tuple.n_node)
      num_edges = tf.reduce_sum(graphs_tuple.n_edge)
      num_graphs = tf.shape(graphs_tuple.n_node)[0]
      return GraphsTupleSize(num_nodes, num_edges, num_graphs)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  import functools
  from absl import logging
  from graph_nets import graphs
  from graph_nets import utils_np
  import six
  from six.moves import range
  import tensorflow
  import tree
label_desc: Use image node_id 33 for calling a global function with example usage: get_graphs_tuple_size(graphs_tuple) and returns: GraphsTupleSize