From e414d6025d8b777cca220c90dc539ef062cbf268 Mon Sep 17 00:00:00 2001 From: yixin <2050485123@qq.com> Date: Wed, 25 Dec 2024 09:28:55 +0800 Subject: [PATCH] add comments for _extends\graph_kernel\expanders\addn.py --- .../_extends/graph_kernel/expanders/addn.py | 27 +++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/src/mindspore2022/mindspore/python/mindspore/_extends/graph_kernel/expanders/addn.py b/src/mindspore2022/mindspore/python/mindspore/_extends/graph_kernel/expanders/addn.py index 8dd9049c..111d637f 100644 --- a/src/mindspore2022/mindspore/python/mindspore/_extends/graph_kernel/expanders/addn.py +++ b/src/mindspore2022/mindspore/python/mindspore/_extends/graph_kernel/expanders/addn.py @@ -13,20 +13,47 @@ # limitations under the License. # =========================================================================== """generate json desc for addn""" +# 导入GraphKernelUnsupportedException异常类 from mindspore._extends.graph_kernel.model.model import GraphKernelUnsupportedException as GKException +# 导入Expander和ExpanderInfoValidator类 from ._utils import Expander, ExpanderInfoValidator as VLD +# 使用VLD.check_all_formats_same装饰器,确保所有输入格式相同 @VLD.check_all_formats_same class AddN(Expander): """Expand AddN to multiple Adds""" + # 检查输入数量是否大于1 def _check(self): + """ + 检查输入的数量是否满足要求。 + + Args: + 无 + + Returns: + 无 + + Raises: + GKException: 如果输入的数量小于2,则抛出GKException异常。 + """ if len(self.inputs) < 2: raise GKException("For 'AddN', the inputs num should be greater than 1, but got {}" .format(len(self.inputs))) + # 将AddN展开为多个Add操作 def _expand(self, graph_builder): + """ + 对输入张量进行逐元素加法运算。 + + Args: + graph_builder (GraphBuilder): 图构建器对象,用于生成图节点。 + + Returns: + Tensor: 逐元素加法运算后的结果张量。 + + """ result = self.inputs[0] for inp in self.inputs[1:]: result = graph_builder.emit('Add', [result, inp])