def _batch_size(self):
    'Extracts the batch size from the graph.'
    return self._session.graph.get_collection('inputs')[0].shape[0].value
Extracts the batch size from the graph.
def _batch_size(self):
    
    return self._session.graph.get_collection('inputs')[0].shape[0].value