Softmax of a tensor.
View aliases
Compat aliases for migration
See Migration guide for more details.
tf.compat.v1.keras.backend.softmax
, `tf.compat.v2.keras.backend.softmax`
tf.keras.backend.softmax(
x, axis=-1
)
Arguments | |
---|---|
x
|
A tensor or variable. |
axis
|
The dimension softmax would be performed on. The default is -1 which indicates the last dimension. |
Returns | |
---|---|
A tensor. |