prelu.js 324 B

12345678910
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. var tf = require("@tensorflow/tfjs-core");
  4. function prelu(x, alpha) {
  5. return tf.tidy(function () {
  6. return tf.add(tf.relu(x), tf.mul(alpha, tf.neg(tf.relu(tf.neg(x)))));
  7. });
  8. }
  9. exports.prelu = prelu;
  10. //# sourceMappingURL=prelu.js.map