I am building a VAE loading huge dataset. The input data is 3D binary voxel data having the dimension like (batch_size, 48 , 48, 48). To load the data one by one in training, I
