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.enable_nested_tensor = enable_nested_tensor
|
||||||
self.mask_check = mask_check
|
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.
|
r"""Pass the input through the encoder layers in turn.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user