code: def _get_inner_node_mask(graph, ntype_id):
      if NTYPE in graph.ndata:
          dtype = F.dtype(graph.ndata["inner_node"])
          return (
              graph.ndata["inner_node"]
              * F.astype(graph.ndata[NTYPE] == ntype_id, dtype)
              == 1
          )
      else:
          return graph.ndata["inner_node"] == 1
  
imports: import json
  import logging
  import os
  import time
  import numpy
  import torch
  from ..None import backend
  from ..base import DGLError, EID, ETYPE, NID, NTYPE
  from ..convert import to_homogeneous
  from ..data.utils import load_graphs, load_tensors, save_graphs, save_tensors
  from ..partition import get_peak_mem, metis_partition_assignment, partition_graph_with_halo
  from ..random import choice
  from ..transforms import sort_csc_by_tag, sort_csr_by_tag
  from .constants import DEFAULT_ETYPE, DEFAULT_NTYPE
  from .graph_partition_book import _etype_str_to_tuple, _etype_tuple_to_str, RangePartitionBook
label_desc: Use image node_id 5 for calling a global function with example usage: _get_inner_node_mask(graph, ntype_id) and returns: unknown, unknown