awslabs/gluonts

`lagged_sequence_values` helper does not work well with multivariate data

Open

#2,708 opened on Mar 6, 2023

View on GitHub
 (7 comments) (0 reactions) (0 assignees)Python (753 forks)batch import
enhancementgood first issuemultivariate

Repository metrics

Stars
 (3,888 stars)
PR merge metrics
 (Avg merge 26d 17h) (4 merged PRs in 30d)

Description

one issue i found with the lagged_sequence_values helper is that with multivariate inputs this helper doesn't return the stacked tensors... e.g.

# prior_input.shape torch.Size([1, 167, 137])
# input.shape torch.Size([1, 24, 137])

lags = lagged_sequence_values([0, 167], prior_input, input, dim=1)
# lags.shape torch.Size([1, 24, 137, 2])

where i would have expected torch.Size([1, 24, 137*2])

Originally posted by @kashif in https://github.com/awslabs/gluonts/issues/2618#issuecomment-1455833300

Contributor guide