Skip to content

Commit 924fe5f

Browse files
committed
Add a pytorch tensor representation type
Signed-off-by: Jibin Varghese <[email protected]>
1 parent 3030fa9 commit 924fe5f

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
from _pydevd_bundle.pydevd_extension_api import StrPresentationProvider
2+
from .pydevd_helpers import find_mod_attr, find_class_name
3+
4+
5+
class PyTorchTensorFormStr:
6+
def can_provide(self, type_object, type_name):
7+
torch_tensor_class = find_mod_attr('torch', 'Tensor')
8+
return torch_tensor_class is not None and issubclass(type_object, torch_tensor_class)
9+
10+
def get_str(self, val):
11+
return "torch.Tensor [ %s , %s ]: %r" % (val.shape, val.device, val)
12+
13+
14+
import sys
15+
16+
if not sys.platform.startswith("java"):
17+
StrPresentationProvider.register(PyTorchTensorFormStr)

0 commit comments

Comments
 (0)