Check if we can switch to monai's `decollate_batch` now https://github.com/ganslate-team/ganslate/blob/14a6655ed2061d660c74804ab92fc463b6225528/ganslate/utils/io.py#L111