View source on GitHub |
Applies an activation function to an output.
tf.keras.layers.Activation( activation, **kwargs )
Arguments | |
---|---|
activation | Activation function, such as tf.nn.relu , or string name of built-in activation function, such as "relu". |
layer = tf.keras.layers.Activation('relu') output = layer([-3.0, -1.0, 0.0, 2.0]) list(output.numpy()) [0.0, 0.0, 0.0, 2.0] layer = tf.keras.layers.Activation(tf.nn.relu) output = layer([-3.0, -1.0, 0.0, 2.0]) list(output.numpy()) [0.0, 0.0, 0.0, 2.0]
Arbitrary. Use the keyword argument input_shape
(tuple of integers, does not include the batch axis) when using this layer as the first layer in a model.
Same shape as input.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/keras/layers/Activation