scatterNd传入三个参数，返回的是向一个新的空Tensor上添加值。

scatter则是在传入的Tensor基础上添加值。

而且添加值的方法也不一样。

请问有什么方法可以解决，或者提供一种自定义实现类似PyTorch中scatter的思路

****************************************************解答*****************************************************

MindSpore中的scatterNd算子原理如下

根据指定的索引将更新值散布到新Tensor上。

https://mindspore.cn/docs/zh-CN/master/api_python/ops/mindspore.ops.ScatterNd.html#mindspore.ops.ScatterNd

或者参考自定义算子教程试试呢

https://mindspore.cn/tutorials/experts/zh-CN/master/operation/op_custom.html

# pytorch # 人工智能 # python
Logo
昇腾开源生态专区

昇腾计算产业是基于昇腾系列（HUAWEI Ascend）处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务，https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐

cover

DeepSeek 崩了 13 小时，不是故障，是 V4 在换引擎

avatar 昇腾开源生态专区

体系结构论文（107）：AscendOptimizer: Episodic Agent for Ascend NPU Operator Optimization

摘要：本文提出AscendOptimizer系统，针对华为Ascend NPU的AscendC算子优化难题，通过双阶段交替优化方法实现性能提升。系统将算子拆分为host侧tiling program和device侧kernel program：Stage I采用进化搜索优化tiling策略，利用硬件反馈探索可行解空间；Stage II通过"优化回退"机制从优质kernel反向构

avatar 昇腾开源生态专区

AtomGit模型托管与实验管理全指南

AtomGit模型托管指南：AI开发者的版本控制利器 摘要：本文介绍AtomGit平台针对AI开发者推出的模型托管与实验管理功能，解决传统Git无法有效管理大模型文件、实验参数分散等问题。AtomGit通过Git LFS大文件存储、模型卡片元数据记录、代码与模型版本关联等创新功能，实现AI项目的一体化管理。文章详细演示了如何创建模型仓库、配置Git LFS管理大文件、编写结构化模型卡片，并特别介绍

avatar 昇腾开源生态专区