View source on GitHub
|
Log-softmax activation function.
tf.keras.ops.log_softmax(
x, axis=-1
)
It is defined as:
f(x) = x - max(x) - log(sum(exp(x - max(x))))
Args | |
|---|---|
x
|
Input tensor. |
axis
|
Integer, axis along which the log-softmax is applied.
Defaults to -1.
|
Returns | |
|---|---|
A tensor with the same shape as x.
|
Example:
x = np.array([-1., 0., 1.])x_log_softmax = keras.ops.log_softmax(x)print(x_log_softmax)array([-2.40760596, -1.40760596, -0.40760596], shape=(3,), dtype=float64)