2885 2899 Add support to support subclass of Tensor or numpy array (#2900)

* [DLMED] fix type issue

Signed-off-by: Nic Ma <[email protected]>

* [DLMED] fix test

Signed-off-by: Nic Ma <[email protected]>

* [DLMED] simplify the change

Signed-off-by: Nic Ma <[email protected]>

* [DLMED] fix flake8

Signed-off-by: Nic Ma <[email protected]>

Author: Nic-Ma

monai/utils/type_conversion.py

@@ -171,7 +171,14 @@ def convert_data_type(
     Returns:
         modified data, orig_type, orig_device
     """
-    orig_type = type(data)
+    orig_type: Any
+    if isinstance(data, torch.Tensor):
+        orig_type = torch.Tensor
+    elif isinstance(data, np.ndarray):
+        orig_type = np.ndarray
+    else:
+        orig_type = type(data)
+
     orig_device = data.device if isinstance(data, torch.Tensor) else None
 
     output_type = output_type or orig_type