import comment and function of op print
This commit is contained in:
parent
022c79d34e
commit
0eb72d76f0
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue