SoftMax函数

SoftMax函数主要用于神经网络输出层设计,通过纳皮尔常数e的指数形式对输出层原始数组进行转换,对转换后数组求和,计算每个元素与数组求和后所占比例。由于softmax输出数组中所有元素之和为1,因此输出层使用softmax函数进行转换后相当于求每个元素概率。

python实现:

import numpy as np
def softmax(a):
    c = np.max(a)
    exp_a = np.exp(a - c)   #防止溢出
    sum_exp_a = np.sum(exp_a)
    y = exp_a / sum_exp_a
    return y

java实现:

public double[] softMax(double[] arr) {
		int length = arr.length;
		Arrays.sort(arr);
		double max = arr[length - 1];
		double[] exp_a = new double[arr.length];
		for(int i = 0; i < length; i++) {
			exp_a[i] = Math.pow(Math.E, arr[i] - max);
		}
		double sum = 0;
		for(int i = 0; i < length; i++) {
			sum += exp_a[i];
		}
		double[] result = new double[length];
		for(int i = 0; i < length; i++) {
			result[i] = exp_a[i] / sum;
		}
		return result;
	}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

TheMatrixs

你的鼓励将是我创作的最大动力!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值