def build_step(self, signals):
j = signals.gather(self.J_data)
j -= self.one
z = tf.nn.softplus(j / self.sigma) * self.sigma
z += self.epsilon
rates = self.amplitude / (
self.tau_ref + self.tau_rc * tf.log1p(tf.reciprocal(z)))
signals.scatter(self.output_data, rates)
评论列表
文章目录