如何在 conv2d 層中使用 PReLU

Question

我正在嘗試在下面的代碼行中使用 PReLU 激活 function，請告訴我該怎么做。

x = Conv2D(64, (3,3), padding='same', strides= (2,2), "activation = tf.keras.activations.sigmoid", name='layer1')(x)
x = MaxPooling2D(pool_size=(2,2))(x)

Answer 1

您可以將其作為層傳遞，如下所示：

x = Conv2D(64, (3,3), padding='same', strides= (2,2))(x)
x = tf.keras.layers.PReLU()(x)
x = MaxPooling2D(pool_size=(2,2))(x)

如文檔中所述