SoftMax激活函数的Python实现

最新推荐文章于 2022-10-09 13:00:00 发布

heze09

最新推荐文章于 2022-10-09 13:00:00 发布

阅读量2k

点赞数

文章标签： python tensorflow

本文链接：https://blog.csdn.net/heze09/article/details/123943800

版权

话不多说，直接上代码

import numpy
import math

def Softmax(inMatrix):
    # Use a breakpoint in the code line below to debug your script.
    # print(f'Hi, {name}')  # Press Ctrl+F8 to toggle the breakpoint.
    m, n = numpy.shape(inMatrix)
    OutMatrix = numpy.mat(numpy.zeros((m, n)))
    soft_sum = 0
    for idx in range(0, n):
        OutMatrix[0, idx] = math.exp(inMatrix[0, idx])
        soft_sum += OutMatrix[0, idx]
    for idx in range(0, n):
        OutMatrix[0, idx] = OutMatrix[0, idx] / soft_sum
    return OutMatrix

# Press the green button in the gutter to run the script.
if __name__ == '__main__':
    a = numpy.array([[1, 2, 3, 2, 1, 4, 3]])
    b = Softmax(a)
    print(b)