code: def pre_compute(A, X, k):
      for _ in range(k):
          X = A @ X
      return X
  
file_path: dgl/examples/sparse/sgc.py
imports: import dgl.sparse
  import torch
  import torch.nn
  import torch.nn.functional
  from dgl.data import CoraGraphDataset
  from torch.optim import Adam
label_desc: Use image node_id 1 for calling a global function with example usage: pre_compute(A, X, k) and returns: X