Now, post processing transforms assume the input to be channel first tensors, but in: https://github.com/Project-MONAI/MONAI/blob/dev/tests/test_integration_segmentation_3d.py#L149 It used batch first tensors.