【Java】Java自定义Softmax函数

最近在Java上想做Softmax，发现居然没有原生的函数支持，于是自己借鉴实现了一个如下：

private static double[] softmax(double[] input) {
    double total = Arrays.stream(input).map(Math::exp).sum();
    double[] output = new double[input.length];
    for (int i=0; i<input.length; i++) {
        output[i] = Math.exp(input[i]) / total;
    }
    return output;
}

猜你喜欢：👇🏻
⭐【Java】列表那些入门的list操作方法