tf.layers.SoftmaxLayer module
tf.layers.SoftmaxLayer module¶
- class tf.layers.SoftmaxLayer.SoftmaxLayer(*args: Any, **kwargs: Any)¶
Bases:
modelzoo.common.tf.layers.BaseLayer.BaseLayerWrapper around the Keras softmax layer.
- call(inputs, **kwargs)¶
Performs the softmax.
- Parameters
inputs – Arbitrary tensor.
- Returns
A tensor of the same shape as input.
- Return type
Tensor