Save the date! Google I/O returns May 18-20 Register now


TensorFlow 2 version View source on GitHub

Applies an activation function to an output.

Inherits From: Layer

activation Activation function, such as tf.nn.relu, or string name of built-in activation function, such as "relu".

Input shape:

Arbitrary. Use the keyword argument input_shape (tuple of integers, does not include the samples axis) when using this layer as the first layer in a model.

Output shape:

Same shape as input.