'Args:
hdf_filename (string): Path to the HDF5 datafile.
name (string, optional): Name to assign this iterator. Defaults to None.'

def __init__(self, hdf_filename, name=None):

    super(HDF5IteratorOneHot, self).__init__(hdf_filename, name=name)
    if ('output' in self.hdf_file):
        assert ('nclass' in self.hdf_file['output'].attrs), 'Missing    nclass    attribute'
        self.nclass = int(self.hdf_file['output'].attrs['nclass'])
        self.out = np.array(self.hdf_file['output'], dtype=np.int32).reshape(((-1), 1))
