code: def unmap(data, count, inds, fill=0):
      """Unmap a subset of item (data) back to the original set of items (of size
      count)"""
      if data.dim() == 1:
          ret = data.new_full((count,), fill)
          ret[inds.type(torch.bool)] = data
      else:
          new_size = (count,) + data.size()[1:]
          ret = data.new_full(new_size, fill)
          # print(inds)
          # print('CCC', ret.shape, inds.shape, data.shape)
          ret[inds.type(torch.bool), :] = data
      return ret
  
imports: from functools import partial
  import torch
  from six.moves import map, zip
label_desc: Use image node_id 2 for calling a global function with example usage: unmap(data, count, inds, fill) and returns: ret