Remove redundant decollate condition for torch/numpy scalars

Signed-off-by: Arthur Dujardin <[email protected]>

Author: arthurdjn

monai/data/utils.py

@@ -625,13 +625,12 @@ def decollate_batch(batch, detach: bool = True, pad=True, fill_value=None):
         type(batch).__module__ == "numpy" and not isinstance(batch, Iterable)
     ):
         return batch
+    # if scalar tensor/array, return the item itself.
     if getattr(batch, "ndim", -1) == 0 and hasattr(batch, "item"):
         return batch.item() if detach else batch
     if isinstance(batch, torch.Tensor):
         if detach:
             batch = batch.detach()
-        if batch.ndim == 0:
-            return batch.item() if detach else batch
         out_list = torch.unbind(batch, dim=0)
         # if of type MetaObj, decollate the metadata
         if isinstance(batch, MetaObj):