Add a pytorch tensor representation type

Signed-off-by: Jibin Varghese <[email protected]>

Author: codeJRV

pydevd_plugins/extensions/types/pydevd_plugin_pytorch_types.py

@@ -0,0 +1,19 @@
+from _pydevd_bundle.pydevd_extension_api import StrPresentationProvider
+from .pydevd_helpers import find_mod_attr, find_class_name
+
+
+class PyTorchTensorFormStr:
+    def can_provide(self, type_object, type_name):
+        torch_tensor_class = find_mod_attr('torch', 'Tensor')
+        return torch_tensor_class is not None and issubclass(type_object, torch_tensor_class)
+
+    def get_str(self, val):
+        if hasattr(val, 'shape') and hasattr(val, 'device'):
+            return "%s [shape: %s, device: %s]: %r" % (find_class_name(val), str(list(val.shape)), val.device, val)
+        else:
+            return "%s: %r" % (find_class_name(val), val)
+
+import sys
+
+if not sys.platform.startswith("java"):
+    StrPresentationProvider.register(PyTorchTensorFormStr)