sequence

sequence_mask(seq_ids, valid_lengths)[源代码]

To boost the performance, this sequence_mask is different with paddle.fluid.layers.sequence_mask

参数
  • seq_ids (Tensor) -- The whole sequence index, a tensor with a shape of [batch_size, sequence_length].

  • valid_lengths (Tensor) -- The valid length of every sequence, a tensor with a shape of [batch_size].

返回

Returns the output sequence mask mask. Its dtype is bool and has a shape of [batch_size, sequence_length].

返回类型

Tensor