wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-14 11:17:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

forgot unsqueeze()

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2019-11-19 14:55:52 +01:00
parent 563ef86436
commit d7aef183ce
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/deepar/deepar_network.py
+1 -1
View File
@@ -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)