class_docstr: Dataset-like object, but only access a subset of it. 
code: def __init__(self, data, index):
      self.data = data
      self.index = index
  
imports: import os
  import torch
  import torch.distributed
  import torch.nn
  import torch.nn.functional
  import torch.optim
  import nni
  import logging
  from math import ceil
  from random import Random
  from torch.autograd import Variable
  from torchvision import datasets, transforms
label_desc: Use image node_id 1 to create a new Partition object from inherited base classes: object with example: obj = Partition(data, index)