mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Update decoder.py
This commit is contained in:
parent
c24b610389
commit
834114f1c2
@ -76,8 +76,8 @@ class Decoder(nn.Module):
|
||||
y:
|
||||
A 2-D tensor of shape (N, U) with blank prepended.
|
||||
need_pad:
|
||||
True to left padding the input. Should be True during training.
|
||||
False to not pad the input. Used only during inference.
|
||||
True to left pad the input. Should be True during training.
|
||||
False to not pad the input. Should be False during inference.
|
||||
Returns:
|
||||
Return a tensor of shape (N, U, embedding_dim).
|
||||
"""
|
||||
|
Loading…
x
Reference in New Issue
Block a user