code: def get_binary_mask(total_size, indices):
      mask = torch.zeros(total_size)
      mask[indices] = 1
      return mask.bool()
  
file_path: AutoGL/autogl/datasets/_heterogeneous_datasets/_dgl_heterogeneous_datasets.py
imports: import os
  import dgl.data.utils
  import numpy
  import scipy.io
  import torch
  from autogl.data import InMemoryStaticGraphSet
  from ..None import _dataset_registry
label_desc: Use image node_id 1 for calling a global function with example usage: get_binary_mask(total_size, indices) and returns: mask