The error I am getting
[original->remapped]: (2552,4,90)->(2552,newaxis,newaxis) (2603,90,1)->(2603,newaxis,newaxis) and requested shape (4,1)