espnet2.asr_transducer.normalization.BasicNorm

source

class espnet2.asr_transducer.normalization.BasicNorm(normalized_shape: int, eps: float = 0.25)

Bases: Module

BasicNorm module definition.

Reference: https://github.com/k2-fsa/icefall/pull/288

• Parameters:
  • normalized_shape – Expected size.
  • eps – Value added to the denominator for numerical stability.

Construct a BasicNorm object.

forward(x: Tensor) → Tensor

Compute basic normalization.

• Parameters:x – Input sequences. (B, T, D_hidden)
• Returns: Output sequences. (B, T, D_hidden)