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
784ab0cfb1
commit
909ec64842
Binary file not shown.
@ -409,7 +409,7 @@ class TransformerEncoder(Module):
|
||||
self.enable_nested_tensor = enable_nested_tensor
|
||||
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:
|
||||
"""Pass the input through the encoder layers in turn.
|
||||
|
||||
Args:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user