concat_sequences

pytorch_forecasting.utils._utils.concat_sequences(sequences: list[Tensor] | list[PackedSequence]) → Tensor | PackedSequence

Concatenate RNN sequences.

Parameters:

sequences (list of torch.Tensor or list of rnn.PackedSequence) – List of RNN packed sequences or tensors of which first index are samples and second are timesteps.

Returns:

Concatenated sequence.

Return type:

torch.Tensor or rnn.PackedSequence