连接序列#

pytorch_forecasting.utils._utils.concat_sequences(sequences: List[Tensor] | List[PackedSequence]) Tensor | PackedSequence[来源]#

连接 RNN 序列。

Parameters:

sequences (Union[List[torch.Tensor], List[rnn.PackedSequence]) – RNN 压缩序列或张量的列表,其中第一个索引是样本,第二个是时间步

Returns:

连接序列

Return type:

Union[torch.Tensor, rnn.PackedSequence]