Registration is open for TensorFlow Dev Summit 2020 Learn more


TensorFlow 1 version View source on GitHub

Class Softmax

Softmax activation function.

Inherits From: Layer

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 the input.


  • axis: Integer, axis along which the softmax normalization is applied.


View source


Compat aliases