wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-08-03 13:10:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix repeated hidden state

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2019-12-09 14:49:12 +01:00
parent f1c00f3780
commit db6a548a02
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/deepar/deepar_network.py
+2 -3
View File
@@ -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
]