tensorflow添加神經(jīng)層

# 添加神經(jīng)層
# add layer
# 層包括weights,biases,激勵函數(shù)
import tensorflow as tf

# 添加層,有輸入,輸入輸出大小,激勵函數(shù)(默認(rèn)是為None,則是線性函數(shù))
def add_layer(inputs,in_size,out_size,activation_function=None):
    Weights = tf.Variable(tf.random_normal([in_size,out_size])) # 權(quán)重 用隨機(jī)變量會比全部為0好很多 ,這里未變量矩陣
    biases = tf.Variable(tf.zeros([1,out_size]) + 0.1) # biases不是矩陣,是類似列表的東西,在激勵學(xué)習(xí)中,biases推薦不為0
    Wx_plus_b = tf.matmul(inputs,Weights) + biases # 矩陣乘法 這是預(yù)測的值,還沒被激活
    if activation_function is None:
        outputs = Wx_plus_b # 如果是線性函數(shù)則不需要添加激勵函數(shù)(因?yàn)榧詈瘮?shù)是非線性方程)
    else:
        outputs = activation_function(Wx_plus_b) # 把Wx_plus_b激活
    return outputs
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容