fastNLP.modules.encoder.seq2seq_encoder module

undocumented

class fastNLP.modules.encoder.seq2seq_encoder.Seq2SeqEncoder[源代码]

基类:torch.nn.modules.module.Module

别名 fastNLP.modules.Seq2SeqEncoder fastNLP.modules.encoder.Seq2SeqEncoder

所有Sequence2Sequence Encoder的基类。需要实现forward函数

forward(tokens, seq_len)[源代码]
参数
  • tokens (torch.LongTensor) – bsz x max_len, encoder的输入

  • seq_len (torch.LongTensor) – bsz

返回

training: bool