code: def _fetch_cuda(indices, tensor, device, **kwargs):
      return torch.index_select(tensor, 0, indices).to(device, **kwargs)
  
file_path: dgl/python/dgl/storages/pytorch_tensor.py
imports: import torch
  from ..utils import gather_pinned_tensor_rows
  from .base import register_storage_wrapper
  from .tensor import BaseTensorStorage
label_desc: Use image node_id 2 for calling a global function with example usage: _fetch_cuda(indices, tensor, device) and returns: torch