提交 0eb72d76 编写于 作者: G guohongzilong

import comment and function of op print

上级 022c79d3
......@@ -513,7 +513,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);
......
......@@ -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):
>>> self.print(x)
>>> def construct(self, x, y):
>>> 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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册