code: def compute_stats(train_curr):
      data = np.vstack(train_curr).reshape(-1, fea_num)
      stat_median = np.median(data, axis=0)
      stat_max = np.quantile(data, 0.95, axis=0)
      stat_min = np.quantile(data, 0.05, axis=0)
      return stat_median, stat_max, stat_min
  
file_path: dgl/examples/pytorch/graphsim/n_body_sim.py
imports: from __future__ import absolute_import, division, print_function
  import argparse
  import os
  from math import cos, pi, radians, sin
  import numpy
label_desc: Use image node_id 4 for calling a global function with example usage: compute_stats(train_curr) and returns: stat_median,  stat_max,  stat_min