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