@thi.ng/tensors/softmax.js

0D/1D/2D/3D/4D tensors with extensible polymorphic operations and customizable storage

import { divN } from "./divn.js";
import { exp } from "./exp.js";
import { sum } from "./sum.js";
const softMax = (out, src, temperature) => {
  const e = temperature !== void 0 ? exp(null, divN(src.empty(), src, temperature)) : exp(src.empty(), src);
  return divN(out || src, e, sum(e));
};
export {
  softMax
};