mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
f6512e7226
commit
7cc43f7bf6
Binary file not shown.
@ -410,7 +410,7 @@ class TransformerEncoder(Module):
|
|||||||
self.mask_check = mask_check
|
self.mask_check = mask_check
|
||||||
|
|
||||||
def forward(self, src: Tensor, mask: Optional[Tensor] = None, src_key_padding_mask: Optional[Tensor] = None) -> Tensor:
|
def forward(self, src: Tensor, mask: Optional[Tensor] = None, src_key_padding_mask: Optional[Tensor] = None) -> Tensor:
|
||||||
r"""Pass the input through the encoder layers in turn.
|
"""Pass the input through the encoder layers in turn.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
src: the sequence to the encoder (required).
|
src: the sequence to the encoder (required).
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user