From e050dd6dc9fe7c3ad846645563e2cf94fde5dbfd Mon Sep 17 00:00:00 2001 From: lzhengning Date: Fri, 16 Sep 2022 16:44:43 +0800 Subject: [PATCH] pretty print jt.Var --- python/jittor/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/jittor/__init__.py b/python/jittor/__init__.py index a8ed991f..2fa3af44 100644 --- a/python/jittor/__init__.py +++ b/python/jittor/__init__.py @@ -1710,7 +1710,9 @@ def jittor_exit(): atexit.register(jittor_exit) def vtos(v): - return f"jt.Var({v.data}, dtype={v.dtype})" + data_str = f"jt.Var({v.data}, dtype={v.dtype})" + data_str = data_str.replace("\n", "\n ") + return data_str Var.__str__ = vtos Var.__repr__ = vtos