TensorFlow 1 version | View source on GitHub |
Softmax of a tensor.
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. |