I’ve looked at it as well as the pytorch-forecasting 4 implementation but I’m not sure I get it because there’s manipulation for the shapes which don’t make sense to me such as unsqueezing the result.