1, "dimension must be larger than 1") # print(np.max(x, axis = 1, keepdims = True)) # axis = 1..._np.max(x, axis=self.dim, keepdims=true)">
赞
踩
- import numpy as np
-
- def softmax(x):
- """ softmax function """
-
- # assert(len(x.shape) > 1, "dimension must be larger than 1")
- # print(np.max(x, axis = 1, keepdims = True)) # axis = 1, 行
-
- x -= np.max(x, axis = 1, keepdims = True) #为了稳定地计算softmax概率, 一般会减掉最大的那个元素
-
- print("减去行最大值 :\n", x)
-
- x = np.exp(x) / np.sum(np.exp(x), axis = 1, keepdims = True)
-
- return x
-
- x = np.random.randint(low = 1, high = 5, size = (2, 3)) #生成一个2x3的矩阵,取值范围在1-5之间
- print("原始 :\n", x)
-
- x_ = softmax(x)
- print("变换后 :\n", x_)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。