src.model.deeplearn.layer.torf_reduce_max_layer

Classes

TORFReduceMaxLayer(*args, **kwargs)

class src.model.deeplearn.layer.torf_reduce_max_layer.TORFReduceMaxLayer(*args, **kwargs)

Author:

Alberto M. Esmoris Pena

Reduce a tensor by taking the maximum along a given axis.

\[f(\mathbf{X}) = \max_{\text{axis}} \mathbf{X}\]

Variables:

axis (int) – The axis along which to reduce.

__init__(axis=1, **kwargs)

See Layer and Layer.__init__().

call(inputs, training=False, mask=False)

Take the max of the input tensor along the configured axis.

Parameters:

inputs – The input tensor.

Returns:

The reduced tensor.

get_config()

Return necessary data to deserialize the layer.

classmethod from_config(config)

Use given config data to deserialize the layer.