[master] update printing

Author: vacancy

jacinle/__init__.py

@@ -49,7 +49,7 @@
     from jacinle.utils.naming import class_name, func_name, method_name, class_name_of_method
     from jacinle.utils.network import get_local_addr
     from jacinle.utils.numeric import safe_sum, mean, std, rms, prod, divup
-    from jacinle.utils.printing import indent_text, stprint, stformat, kvprint, kvformat, print_to_string
+    from jacinle.utils.printing import indent_text, stprint, stformat, kvprint, kvformat, print_to_string, print_to
     from jacinle.utils.tqdm import get_current_tqdm, tqdm, tqdm_pbar, tqdm_gofor, tqdm_zip
 
     from jacinle import io

jacinle/utils/printing.py

@@ -12,6 +12,7 @@
 import sys
 import numpy as np
 import collections
+import contextlib
 
 import threading
 from .registry import LockRegistry
@@ -194,6 +195,13 @@ def print_to_string(target='STDOUT'):
     return PrintToStringContext(target, need_lock=True)
 
 
+@contextlib.contextmanager
+def print_to(print_func, target='STDOUT'):
+    with PrintToStringContext(target, need_lock=True) as ctx:
+        yield
+    print_func(ctx.get())
+
+
 def print2format(print_func):
     def format_func(*args, **kwargs):
         f = io.StringIO()

jactorch/nn/cnn/functional.py

@@ -22,6 +22,7 @@ class ConvPaddingMode(JacEnum):
     DEFAULT = 'default'
     VALID = 'valid'
     SAME = 'same'
+    TENSORFLOW = 'tensorflow'
 
 
 class ConvBorderMode(JacEnum):