What should be input shape be?

Hi @pablowilks,

I have answer a similar question here. I hope it helps.
Best,