torch.nn.utils.unpad_sequence
function unpad_sequence(padded_sequence: Tensor, lengths: number[], options?: UnpadSequenceOptions): Tensor[]function unpad_sequence(padded_sequence: Tensor, lengths: number[], batch_first: boolean, options?: UnpadSequenceOptions): Tensor[]Unpad a padded Tensor into a list of variable length Tensors.
Parameters
padded_sequenceTensor- Padded sequence tensor
lengthsnumber[]- Length of each sequence in the batch
optionsUnpadSequenceOptionsoptional- Optional settings for unpadding
Returns
Tensor[]– List of tensors