I’m sure I understand the issue completely, as you are explaining to work with 2D data but also with 4D tensors. In the former case:



 Adrien88:

However, in my case, my Input_Dimension is 2D, which is (rows x columns)
rows: number of rows in one day, which is 1782
columns: number of features, which is 2 (store and family)


it seems that the batch dimension might be missing and you should be able to add it into dim0.