how can I shape this to 4D in python-3.x
--------------------------------------------------------------------------- ValueError Tra