sequence

sequence_mask(seq_ids, valid_lengths)

To boost the performance, this sequence_mask is different with paddle.nn.functional.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