關(guān)于tensorflow softmax函數(shù)用法解析
如下所示:
def softmax(logits, axis=None, name=None, dim=None): """Computes softmax activations. This function performs the equivalent of softmax = tf.exp(logits) / tf.reduce_sum(tf.exp(logits), axis) Args: logits: A non-empty `Tensor`. Must be one of the following types: `half`, `float32`, `float64`. axis: The dimension softmax would be performed on. The default is -1 which indicates the last dimension. name: A name for the operation (optional). dim: Deprecated alias for `axis`. Returns: A `Tensor`. Has the same type and shape as `logits`. Raises: InvalidArgumentError: if `logits` is empty or `axis` is beyond the last dimension of `logits`. """ axis = deprecation.deprecated_argument_lookup("axis", axis, "dim", dim) if axis is None: axis = -1 return _softmax(logits, gen_nn_ops.softmax, axis, name)
softmax函數(shù)的返回結(jié)果和輸入的tensor有相同的shape,既然沒(méi)有改變tensor的形狀,那么softmax究竟對(duì)tensor做了什么?
答案就是softmax會(huì)以某一個(gè)軸的下標(biāo)為索引,對(duì)這一軸上其他維度的值進(jìn)行 激活 + 歸一化處理。
一般來(lái)說(shuō),這個(gè)索引軸都是表示類(lèi)別的那個(gè)維度(tf.nn.softmax中默認(rèn)為axis=-1,也就是最后一個(gè)維度)
舉例:
def softmax(X, theta = 1.0, axis = None): """ Compute the softmax of each element along an axis of X. Parameters ---------- X: ND-Array. Probably should be floats. theta (optional): float parameter, used as a multiplier prior to exponentiation. Default = 1.0 axis (optional): axis to compute values along. Default is the first non-singleton axis. Returns an array the same size as X. The result will sum to 1 along the specified axis. """ # make X at least 2d y = np.atleast_2d(X) # find axis if axis is None: axis = next(j[0] for j in enumerate(y.shape) if j[1] > 1) # multiply y against the theta parameter, y = y * float(theta) # subtract the max for numerical stability y = y - np.expand_dims(np.max(y, axis = axis), axis) # exponentiate y y = np.exp(y) # take the sum along the specified axis ax_sum = np.expand_dims(np.sum(y, axis = axis), axis) # finally: divide elementwise p = y / ax_sum # flatten if X was 1D if len(X.shape) == 1: p = p.flatten() return p c = np.random.randn(2,3) print(c) # 假設(shè)第0維是類(lèi)別,一共有里兩種類(lèi)別 cc = softmax(c,axis=0) # 假設(shè)最后一維是類(lèi)別,一共有3種類(lèi)別 ccc = softmax(c,axis=-1) print(cc) print(ccc)
結(jié)果:
c: [[-1.30022268 0.59127472 1.21384177] [ 0.1981082 -0.83686108 -1.54785864]] cc: [[0.1826746 0.80661068 0.94057075] [0.8173254 0.19338932 0.05942925]] ccc: [[0.0500392 0.33172426 0.61823654] [0.65371718 0.23222472 0.1140581 ]]
可以看到,對(duì)axis=0的軸做softmax時(shí),輸出結(jié)果在axis=0軸上和為1(eg: 0.1826746+0.8173254),同理在axis=1軸上做的話(huà)結(jié)果的axis=1軸和也為1(eg: 0.0500392+0.33172426+0.61823654)。
這些值是怎么得到的呢?
以cc為例(沿著axis=0做softmax):
以ccc為例(沿著axis=1做softmax):
知道了計(jì)算方法,現(xiàn)在我們?cè)賮?lái)討論一下這些值的實(shí)際意義:
cc[0,0]實(shí)際上表示這樣一種概率: P( label = 0 | value = [-1.30022268 0.1981082] = c[*,0] ) = 0.1826746
cc[1,0]實(shí)際上表示這樣一種概率: P( label = 1 | value = [-1.30022268 0.1981082] = c[*,0] ) = 0.8173254
ccc[0,0]實(shí)際上表示這樣一種概率: P( label = 0 | value = [-1.30022268 0.59127472 1.21384177] = c[0]) = 0.0500392
ccc[0,1]實(shí)際上表示這樣一種概率: P( label = 1 | value = [-1.30022268 0.59127472 1.21384177] = c[0]) = 0.33172426
ccc[0,2]實(shí)際上表示這樣一種概率: P( label = 2 | value = [-1.30022268 0.59127472 1.21384177] = c[0]) = 0.61823654
將他們擴(kuò)展到更多維的情況:假設(shè)c是一個(gè)[batch_size , timesteps, categories]的三維tensor
output = tf.nn.softmax(c,axis=-1)
那么 output[1, 2, 3] 則表示 P(label =3 | value = c[1,2] )
以上這篇關(guān)于tensorflow softmax函數(shù)用法解析就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
PyTorch中關(guān)于tensor.repeat()的使用
這篇文章主要介紹了PyTorch中關(guān)于tensor.repeat()的使用,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2022-11-11python繪圖時(shí),坐標(biāo)軸負(fù)號(hào)顯示不出來(lái)的解決
這篇文章主要介紹了python繪圖時(shí),坐標(biāo)軸負(fù)號(hào)顯示不出來(lái)的解決方案,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2023-09-09python web.py開(kāi)發(fā)httpserver解決跨域問(wèn)題實(shí)例解析
這篇文章主要介紹了python web.py開(kāi)發(fā)httpserver解決跨域問(wèn)題實(shí)例解析,分享了相關(guān)代碼示例,小編覺(jué)得還是挺不錯(cuò)的,具有一定借鑒價(jià)值,需要的朋友可以參考下2018-02-02詳解python執(zhí)行shell腳本創(chuàng)建用戶(hù)及相關(guān)操作
這篇文章主要介紹了python執(zhí)行shell腳本創(chuàng)建用戶(hù)及相關(guān)操作,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-04-04django之從html頁(yè)面表單獲取輸入的數(shù)據(jù)實(shí)例
這篇文章主要介紹了django之從html頁(yè)面表單獲取輸入的數(shù)據(jù)實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-03-03利用一個(gè)簡(jiǎn)單的例子窺探CPython內(nèi)核的運(yùn)行機(jī)制
這篇文章主要介紹了利用一個(gè)簡(jiǎn)單的例子窺探CPython內(nèi)核的運(yùn)行機(jī)制,作者通過(guò)一個(gè)簡(jiǎn)單的輸出函數(shù)深入、介紹了CPython源碼C代碼中的一些函數(shù),需要的朋友可以參考下2015-03-03python連接PostgreSQL數(shù)據(jù)庫(kù)的過(guò)程詳解
這篇文章主要介紹了python連接PostgreSQL數(shù)據(jù)庫(kù)的過(guò)程詳解,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-09-09