code: def get_col_slice(
      x, start_row_idx, end_row_idx, start_col_idx, end_col_idx
  ):
      outs = []
      chunk = 100000
      for i in tqdm(range(start_row_idx, end_row_idx, chunk)):
          j = min(i + chunk, end_row_idx)
          outs.append(x[i:j, start_col_idx:end_col_idx].copy())
      return np.concatenate(outs, axis=0)
  
imports: import os
  import sys
  import yaml
  import pgl
  import time
  import copy
  import numpy
  import os.path
  from pgl.utils.logger import log
  from pgl.graph import Graph
  from pgl import graph_kernel
  from pgl.sampling.custom import subgraph
  from ogb.lsc import MAG240MDataset, MAG240MEvaluator
  import time
  import paddle
  from tqdm import tqdm
  from pgl.utils.helper import scatter
label_desc: Use image node_id 1 for calling a global function with example usage: get_col_slice(x, start_row_idx, end_row_idx, start_col_idx, end_col_idx) and returns: np