pgjmblvch
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add example

This commit is contained in:
lianliguang 2021-06-09 20:35:22 +08:00
parent df491ffe53
commit c5ffb4a774
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mindspore/ops/primitive.py
View File

@ -187,14 +187,19 @@ class Primitive(Primitive_):
Examples:
>>> class AddN(Primitive):
>>> @prim_attr_register
>>> def __init__(self):
>>> self.init_prim_io_names(inputs=["inputs"], outputs=["sum"])
>>> def check_elim(self, inputs):
>>> if len(inputs) != 1:
>>> return (False, None)
>>> if isinstance(inputs[0], Tensor):
>>> return (True, inputs[0])
... @prim_attr_register
... def __init__(self):
... self.init_prim_io_names(inputs=["inputs"], outputs=["sum"])
... def check_elim(self, inputs):
... if len(inputs) != 1:
... 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]))
"""
return (False, None)