tf.layers.LayerNormalizationLayer module

tf.layers.LayerNormalizationLayer module

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

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

Wrapper around the Keras layer normalization. Reference: 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