convert scalar array to tensor

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

Author: arthurdjn

monai/data/utils.py

@@ -626,7 +626,7 @@ def decollate_batch(batch, detach: bool = True, pad=True, fill_value=None):
     ):
         return batch
     if isinstance(batch, np.ndarray) and batch.ndim == 0:
-        return batch.item() if detach else batch
+        batch = torch.from_numpy(batch)
     if isinstance(batch, torch.Tensor):
         if detach:
             batch = batch.detach()