!86 Op Print: add comments and check function

Merge pull request !86 from ghzl/improve-function-and-comment-in-op-print

mindspore/ccsrc/transform/op_adapter.h

@@ -538,7 +538,7 @@ class OpAdapter : public BaseOpAdapter {
         return;
       }
     } else {
-      MS_LOG(ERROR) << "Update output desc failed, unknow output shape type";
+      MS_LOG(WARNING) << "Update output desc failed, unknow output shape type";
       return;
     }
     MS_EXCEPTION_IF_NULL(node);

mindspore/ops/operations/debug_ops.py

@@ -14,6 +14,7 @@
 # ============================================================================
 
 """debug_ops"""
+from ..._checkparam import ParamValidator as validator
 from ...common import dtype as mstype
 from ..primitive import Primitive, prim_attr_register, PrimitiveWithInfer
 
@@ -157,19 +158,20 @@ class InsertGradientOf(PrimitiveWithInfer):
 
 class Print(PrimitiveWithInfer):
     """
-    Output tensor to stdout.
+    Output tensor or string to stdout.
 
     Inputs:
-        - **input_x** (Tensor) - The graph node to attach to.
+        - **input_x** (Union[Tensor, str]) - The graph node to attach to. The input supports
+          multiple strings and tensors which are separated by ','.
 
     Examples:
         >>> class PrintDemo(nn.Cell):
-        >>>     def __init__(self,):
+        >>>     def __init__(self):
         >>>         super(PrintDemo, self).__init__()
         >>>         self.print = P.Print()
         >>>
-        >>>     def construct(self, x):
+        >>>     def construct(self, x, y):
-        >>>         self.print(x)
+        >>>         self.print('Print Tensor x and Tensor y:', x, y)
         >>>         return x
     """
 
@@ -181,4 +183,6 @@ class Print(PrimitiveWithInfer):
         return [1]
 
     def infer_dtype(self, *inputs):
+        for dtype in inputs:
+            validator.check_subclass("input", dtype, (mstype.tensor, mstype.string))
         return mstype.int32