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
bcf241e91d
commit
ca7d41682a
Binary file not shown.
@ -409,7 +409,7 @@ class TransformerEncoder(Module):
|
||||
self.enable_nested_tensor = enable_nested_tensor
|
||||
self.mask_check = mask_check
|
||||
|
||||
[docs] 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.
|
||||
|
||||
Args:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user