解压序列#

pytorch_forecasting.utils._utils.unpack_sequence(sequence: Tensor | PackedSequence) Tuple[Tensor, Tensor][来源]#

解压RNN序列。

Parameters:

sequence (联合[torch.Tensor, rnn.PackedSequence]) – RNN 打包序列或张量,其中第一个索引是样本,第二个是时间步

Returns:

解包序列的元组和样本的长度

Return type:

元组[torch.Tensor, torch.Tensor]