tf.nn.conv2d 和 tf.layers.conv2d

在寫CNN中注意到tensorflow目前有tf.nn.conv2d和tf.layers.conv2d這兩個(gè)很相似的API.

tf.nn.conv2d, 需要自行傳入初始化好的filter(四個(gè)維度),在初始化filter或者說Weights中,已經(jīng)手動(dòng)選擇了初始化方案,一般用的是tf.truncated_normal。另外tf.nn.conv2d中激活函數(shù)需要另外寫。

而在tf.layers.conv2d中,只要寫入n和size,參數(shù)列表中kernel_initializer默認(rèn)是None, 不清楚這里的參數(shù)是如何初始化的? tf.layers.conv2d中,激活函數(shù)是一個(gè)參數(shù),不需要另外調(diào)用。

問過大神,大神答曰:一般推薦使用tf.layers.下面的函數(shù),用起來方便。但是在tf2.0里,tf.layers.下面的API也都被遺棄了,tf2.0推薦使用keras.layers下面的API。

tf.nn.conv2d


tf.layers.conv2d


?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

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

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