From d7aef183ce5a2a533175a87e4657fd4edd5ec399 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Tue, 19 Nov 2019 14:55:52 +0100 Subject: [PATCH] forgot unsqueeze() --- pts/model/deepar/deepar_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pts/model/deepar/deepar_network.py b/pts/model/deepar/deepar_network.py index c9e59df..e9b285c 100644 --- a/pts/model/deepar/deepar_network.py +++ b/pts/model/deepar/deepar_network.py @@ -171,7 +171,7 @@ class DeepARNetwork(nn.Module): ), dim=1) # (batch_size, subsequences_length, num_features + 1) - repeated_static_feat = static_feat.expand(-1, subsequences_length, -1) + repeated_static_feat = static_feat.unsqueeze(1).expand(-1, subsequences_length, -1) # (batch_size, sub_seq_len, *target_shape, num_lags) lags_scaled = lags / scale.unsqueeze(-1)