code: def index2mask(idx: Tensor, size: int) -> Tensor:
      mask = torch.zeros(size, dtype=torch.bool, device=idx.device)
      mask[idx] = True
      return mask
  
file_path: DIG/dig/lsgraph/dataset/get_data.py
imports: from typing import Tuple
  import torch
  from torch import Tensor
  import torch_geometric.transforms
  from torch_geometric.data import Data, Batch
  from torch_geometric.datasets import GNNBenchmarkDataset, Yelp, Flickr, Reddit2
  from ogb.nodeproppred import PygNodePropPredDataset
label_desc: Use image node_id 1 for calling a global function with example usage: index2mask(idx, size) and returns: mask