from local

This commit is contained in:
dohe0342 2023-02-02 13:51:55 +09:00
parent f6512e7226
commit 7cc43f7bf6
2 changed files with 1 additions and 1 deletions

View File

@ -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).