code: def divided_graph(x, batch_index):
      graph = []
      for i in range(batch_index[-1] + 1):
          graph.append(x[batch_index == i])
  
      return graph
  
file_path: DIG/dig/xgraph/models/utils.py
imports: import torch
  from torch_geometric.utils.num_nodes import maybe_num_nodes
label_desc: Use image node_id 2 for calling the ReadOut obj's underlying member method code with example usage: obj.divided_graph(x, batch_index) and returns: graph