Softmax in JS using ndarray

softmax.js

const ndarray = require("ndarray");
const ops = require("ndarray-ops");

// Softmax in-place
const softmax = (() => {
    function denom(arr, C) {
        const tmp = ndarray(new Float32Array(arr.size));
        ops.assign(tmp, arr);
        ops.addseq(tmp, -C);
        ops.expeq(tmp);
        return ops.sum(tmp);
    }

    return (arr) => {
        const C = ops.sup(arr);
        const d = denom(arr, C);
        ops.addseq(arr, -C);
        ops.expeq(arr);
        ops.divseq(arr, d);
    }
})();