Suppose I have a batch of 3D voxel images of shape (batch_size, height, width, depth, channels).
(batch_size, height, width, depth, channels)
v1 = np.expan
