!86 Op Print: add comments and check function

Merge pull request !86 from ghzl/improve-function-and-comment-in-op-print
This commit is contained in:
mindspore-ci-bot 2020-04-03 17:46:42 +08:00 committed by Gitee
commit 97276fa522
2 changed files with 10 additions and 6 deletions

View File

@ -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);

View File

@ -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