ValueError: Dimensions must be equal, but are 2 and 1 in time2vec example

Question

I have 2 inputs and 4 outputs. I want to use the time2vec to predict the outputs. I have used the code in https://towardsdatascience.com/time2vec-for-time-series-features-encod