code: def load_subtensor(g, input_nodes, device):
      """
      Copys features and labels of a set of nodes onto GPU.
      """
      batch_inputs = g.ndata["features"][input_nodes].to(device)
      return batch_inputs
  
imports: import time
  import dgl
  import dgl.function
  import dgl.nn.pytorch
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  import torch.optim
  from ..None import utils
label_desc: Use image node_id 2 for calling a global function with example usage: load_subtensor(g, input_nodes, device) and returns: batch_inputs