code: def batched_assert_zero(tensor, size):
      BATCH_SIZE = 2**16
      curr_pos = 0
      while curr_pos < size:
          end = min(curr_pos + BATCH_SIZE, size)
          assert F.sum(tensor[F.arange(curr_pos, end)], 0) == 0
          curr_pos = end
  
imports: import json
  import os
  from itertools import product
  import dgl
  import dgl.backend
  import numpy
  from dgl.distributed import edge_split, load_partition_book, node_split
label_desc: Use image node_id 1 for calling a global function with example usage: batched_assert_zero(tensor, size) without return types