I was thinking, if i have an input of 128x128x128, this can be represented as 128 128x128 2d arrays or a single 128x128x128 3D array, so which is most effiecent in terms of memory usage and speed on a single GPU?