p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!18350 fix doc 

Merge pull request !18350 from lianliguang/code_docs_primitive
This commit is contained in:
i-robot 2021-06-16 09:29:33 +08:00 committed by Gitee
parent 769c0dd344 4652789670
commit de68aef68c
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mindspore/ops/primitive.py
View File

@ -195,11 +195,12 @@ class Primitive(Primitive_):
... return (False, None)
... if isinstance(inputs[0], Tensor):
... return (True, inputs[0])
...
>>> addn = AddN()
>>> input_x = Tensor(np.array([1, 2, 3]), mindspore.float32)
>>> output = addn.check_elim((input_x,))
>>> print(output)
(True, Tensor(shape = [3], dtype = Float32, value = [1，0000000e+00,2，0000000e+00，3，0000000e+00]))
(True, Tensor(shape = [3], dtype = Float32, value = [1.0000000e+00,2.0000000e+00，3.0000000e+00]))
"""
return (False, None)
@ -506,12 +507,12 @@ def prim_attr_register(fn):
Examples:
>>> class MatMul(PrimitiveWithCheck):
>>> @prim_attr_register
>>> def __init__(self, transpose_a=False, transpose_b=False):
>>> self.init_prim_io_names(inputs=['x1', 'x2'], outputs=['output'])
>>> cls_name = self.name
>>> validator.check_value_type("transpose_a", transpose_a, [bool], cls_name)
>>> validator.check_value_type("transpose_b", transpose_b, [bool], cls_name)
... @prim_attr_register
... def __init__(self, transpose_a=False, transpose_b=False):
... self.init_prim_io_names(inputs=['x1', 'x2'], outputs=['output'])
...
>>> # init a Primitive obj
>>> matmul = MatMul()
"""
def deco(self, *args, **kwargs):