lmcintosh

class AdaptiveIAF(tf.nn.rnn_cell.RNNCell):
    def __init__(self, num_units, dt, reuse=False):
        self._dt = tf.constant(dt, dtype=tf.float32)
        self._num_units = num_units
        self._reuse = reuse
          
    @property
    def state_size(self):
          return (self._num_units, self._num_units)