I am a beginner in python and I faced a problem in maintaining the same size of the last 2 dimensions in the 5D array that I have while looping over it to do some calculatio
