keras-js/lib/activations/elu.js

Run Keras models in the browser, with GPU support using WebGL

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = elu;

var _cwise = _interopRequireDefault(require("cwise"));

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

const _elu = (0, _cwise.default)({
  args: ['array', 'scalar'],
  body: function (_x, alpha) {
    _x = Math.max(_x, 0) + alpha * (Math.exp(Math.min(_x, 0)) - 1);
  }
});

function elu(x, opts = {}) {
  const {
    alpha = 1.0
  } = opts;

  _elu(x.tensor, alpha);
}