add print illustration

4 years ago · 5bd859f82f
parent e706cb2512
commit 5bd859f82f
1 changed files with 4 additions and 0 deletions
--- a/mindspore/ops/operations/debug_ops.py
+++ b/mindspore/ops/operations/debug_ops.py
@ -347,6 +347,10 @@ class Print(PrimitiveWithInfer):
        ...         self.print('Print Tensor x and Tensor y:', x, y)
        ...         return x
        ...
+        >>> x = Tensor(np.ones([2, 1]).astype(np.int32))
+        >>> y = Tensor(np.ones([2, 2]).astype(np.int32))
+        >>> net = PrintDemo()
+        >>> output = net(x, y)
    """

    @prim_attr_register