Update script.js

script.js

@@ -23,7 +23,7 @@ var rand_truncated_normal = function(low, high) {
 
 var softplus = function(x) {
     const m = tf.maximum(x, 0.0);
-    return tf.add(m, tf.log(tf.add(tf.exp(-m), tf.exp(x-m))));
+    return tf.add(m, tf.log(tf.add(tf.exp(tf.neg(m)), tf.exp(tf.sub(x, m)))));
 }