如何在 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)

如文档中所述