Silence an annoying warning (#299)

The new code is taken from here:
https://github.com/pytorch/pytorch/blob/55844dfdbc26383cc03ac212a515257c6a99594c/torch/utils/data/_utils/collate.py#L160

conduit/data/datasets/utils.py

@@ -318,7 +318,10 @@ def _collate(self, batch: Any) -> Any:
                 # If we're in a background process, concatenate directly into a
                 # shared memory tensor to avoid an extra copy
                 numel = sum(x.numel() for x in batch)
-                storage = elem.storage()._new_shared(numel)
+                if hasattr(elem, "_typed_storage"):  # in PyTorch 2.0, it's called _typed_storage
+                    storage = elem._typed_storage()._new_shared(numel, device=elem.device)
+                else:
+                    storage = elem.storage()._new_shared(numel)
                 out = elem.new(storage).resize_(len(batch), *list(elem.size()))
             ndims = elem.dim()
             # If 'batch' is a sequence of sub-batched tensors we concatenate the elements along the