code: def random_walk(
      src: SparseTensor, start: torch.Tensor, walk_length: int
  ) -> torch.Tensor:
      rowptr, col, _ = src.csr()
      return torch.ops.torch_sparse.random_walk(
          rowptr, col, start, walk_length
      )
  
label_desc: Use image node_id 1 for calling a global function with example usage: random_walk(src, start, walk_length) and returns: torch