tf.SoftmaxLayer module
tf.SoftmaxLayer module¶
- class tf.SoftmaxLayer.SoftmaxLayer(*args: Any, **kwargs: Any)¶
Bases:
modelzoo.common.layers.tf.BaseLayer.BaseLayer
Wrapper 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