From db6a548a026f5936d592f312d0638387c87a6ae8 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Mon, 9 Dec 2019 14:49:12 +0100 Subject: [PATCH] fix repeated hidden state --- pts/model/deepar/deepar_network.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pts/model/deepar/deepar_network.py b/pts/model/deepar/deepar_network.py index 1f726e5..0ad7595 100644 --- a/pts/model/deepar/deepar_network.py +++ b/pts/model/deepar/deepar_network.py @@ -331,13 +331,12 @@ class DeepARPredictionNetwork(DeepARNetwork): repeats=self.num_parallel_samples, dim=0 ) repeated_static_feat = static_feat.repeat_interleave( - repeats=self.num_parallel_samples, dim=0 - ).expand_dims(axis=1) + repeats=self.num_parallel_samples, dim=0).unsqueeze(1) repeated_scale = scale.repeat_interleave( repeats=self.num_parallel_samples, dim=0 ) repeated_states = [ - s.repeat_interleave(repeats=self.num_parallel_samples, dim=0) + s.repeat_interleave(repeats=self.num_parallel_samples, dim=1) for s in begin_states ]