日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

tf.nn.conv2d與tf.layers.conv2d的區別及說明_python

作者:zhongranxu ? 更新時間: 2023-05-29 編程語言

tf.nn.conv2d與tf.layers.conv2d的區別

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

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

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

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

tf.nn.conv2d

tf.layers.conv2d

tf.nn.conv2d和tf.layers.conv2d的學習

看了很多tensorflow卷積的例子,有的用了tf.nn.conv2d,有的用了tf.layers.conv2d,那么這兩個究竟有啥不同呢?

tf.layers.conv2d(inputs, filters, kernel_size, strides=(1,1),

???????????????????????padding='valid', data_format='channels_last',
???????????????     dilation_rate=(1,1), activation=None,
???????????????     use_bias=True, kernel_initializer=None,
???????????????     bias_initializer=init_ops.zeros_initializer(),
???????????????     kernel_regularizer=None,
???????????????     bias_regularizer=None,
???????????????     activity_regularizer=None, trainable=True,
???????????????     name=None, reuse=None)

tf.nn.conv2d(input, filter, strides, padding, use_cudnn_on_gpu=None, data_format=None, name=None)

對于卷積來說,作用是一樣的。tf.layers.conv2d 使用tf.nn.convolution作為后端。

一個參數要注意一下:

  • filter:是一個4維張量,其type必須和輸入一樣,
[filter_height, filter_width, in_channels, out_channels]
  • filters:一個整數,輸出空間的維度,也就是卷積核的數量。

tf.layers.conv2d參數豐富,一般用于從頭訓練一個模型。

tf.nn.conv2d,一般在下載預訓練好的模型時使用。

所以,喜歡哪個就用哪個咯。

總結

原文鏈接:https://blog.csdn.net/zhongranxu/article/details/105833035

  • 上一篇:沒有了
  • 下一篇:沒有了
欄目分類
最近更新