unpack_sequence

pytorch_forecasting.utils._utils.unpack_sequence(sequence: Tensor | PackedSequence) → tuple[Tensor, Tensor]

Unpack RNN sequence.

Parameters:

sequence (torch.Tensor or rnn.PackedSequence) – RNN packed sequence or tensor of which first index are samples and second are timesteps.

Returns:

Tuple of unpacked sequence and length of samples.

Return type:

tuple of torch.Tensor