tf.LayerNormalizationLayer module

tf.LayerNormalizationLayer module

class tf.LayerNormalizationLayer.LayerNormalizationLayer(*args: Any, **kwargs: Any)

Bases: modelzoo.common.layers.tf.BaseLayer.BaseLayer

Wrapper around the Keras layer normalization.

call(inputs, **kwargs)

Apply the layer normalization.

Parameters

inputs (Tensor) – Arbitrary tensor.

Returns

A normalized tensor of the same shape as input.

NOTE: While **kwargs are passed, the training arg is never used.

Return type

Tensor

previous

tf.LSTMCell module

next

tf.MaxPool2DLayer module