mog_3d_tensor() in utils.py doesn't adapt to 3D images when W,H,D are not the same.

[One-punch24]

Dear authors,

Thank you for providing this codebase!

It seems that mog_3d_tensor() in utils.py doesn't adapt to 3D images when W,H,D are not the same. It is probably related to the order when you define x,y,z.

Best,
Shuailong