code: def sum(x, dim, keepdims=False):
      return tf.reduce_sum(x, axis=dim, keepdims=keepdims)
  
imports: from __future__ import absolute_import
  import numpy
  import tensorflow
  from scipy.sparse import coo_matrix
label_desc: Use image node_id 17 for calling a global function with example usage: sum(x, dim, keepdims) and returns: tf