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