prelu.js 233 B

1234567
  1. import * as tf from '@tensorflow/tfjs-core';
  2. export function prelu(x, alpha) {
  3. return tf.tidy(function () {
  4. return tf.add(tf.relu(x), tf.mul(alpha, tf.neg(tf.relu(tf.neg(x)))));
  5. });
  6. }
  7. //# sourceMappingURL=prelu.js.map