问TensorFlow能支持尖峰神经元吗？

EN

n = 10
SPIKING_THRESHOLD = 35.0

v = tf.placeholder(tf.float32, shape=[n])
u = tf.placeholder(tf.float32, shape=[n])
i = tf.placeholder(tf.float32)

# Evaluate which neurons have reached the spiking threshold
has_fired_op = tf.greater_equal(v, tf.constant(SPIKING_THRESHOLD, shape=v.shape))

# Evaluate membrane potential increment for the considered time interval
# dv = 0 if the neuron fired, dv = 0.04v*v + 5v + 140 + I -u otherwise
dv_op = tf.where(has_fired_op,
                 tf.zeros(v.shape),
                 tf.subtract(tf.add_n([tf.multiply(tf.square(v), 0.04),
                                       tf.multiply(v, 5.0),
                                       tf.constant(140.0, shape=v.shape),
                                       i]),
                             self.u))

# Evaluate membrane recovery decrement for the considered time interval
# du = 0 if the neuron fired, du = a*(b*v -u) otherwise
du_op = tf.where(has_fired_op,
                 tf.zeros([v.shape]),
                 tf.multiply(A, tf.subtract(tf.multiply(B, v), u)))

with tf.Session() as sess:

    sess.run(tf.global_variables_initializer())

    feed = {u: np.full((n), -13.0), v: np.full((n), -65.0), i : 7.0}

    dv, du = sess.run([dv_op, du_op], feed_dict=feed)