U q±Ëhðã@s>ddlmZddlZddlmZddd„ZGdd„deƒZdS) é)Ú annotationsN)ÚOpRuncCsät |¡}t |jdd…¡D]À}|dkrH|t||ƒ||7<q|dkrn|t||ƒ||9<q|dkrœt |||||¡|t||ƒ<q|dkrÊt |||||¡|t||ƒ<q|||t||ƒ<q|S)NéÿÿÿÿÚaddÚmulÚmaxÚmin)ÚnpÚcopyZndindexÚshapeÚtupleÚmaximumZminimum)ÚdataÚindicesÚupdatesÚ reductionÚoutputÚi©rúC/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_scatternd.pyÚ_scatter_nd_impl s &&rc@seZdZddd„ZdS)Ú ScatterNDNcCst||||d}|fS)N)r)r)ÚselfrrrrÚyrrrÚ_runszScatterND._run)N)Ú__name__Ú __module__Ú __qualname__rrrrrrsr)N)Ú __future__rZnumpyr Zonnx.reference.op_runrrrrrrrÚs