I saw inconsistency in forward_tts _forward_aligner and AlignmentNetwork forward or I am confused!!

[Prajwalbhandari869]

In the docstrings of forward_tts _forward_aligner, it is said the shape of parameter x_mask should be as - x_mask: :math:[B, 1, T_en] and the same shaped x_mask is passed to aligner without squeezing.

But forward of AlignmentNetwork is expecting x_mask as - mask: :math:[B, T_de].

So, what is the actual size expected at AlignmentNetwork forward?

• Should it be actual passed - x_mask: :math:[B, 1, T_en]?
• Or Should it be expecting - mask: :math:[B, T_de]?
• Or is the docstring inconsistent?