sequence

sequence_mask(seq_ids, valid_lengths)[source]

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

Parameters
  • seq_ids (Tensor) – The whole sequence index, a tensor with shape [batch_size, sequence_length].

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

Returns

The output sequence mask.

Return type

Tensor