最近在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操作方法