tf.keras.activations.log_softmax

Log-Softmax activation function.

Each input vector is handled independently. The axis argument sets which axis of the input the function is applied along.

x Input tensor.
axis Integer, axis along which the softmax is applied.