如何解决Numpy 相当于“tf.tensor_scatter_nd_add”方法
问题确实在标题中,我正在寻找 scipy/numpy/etc 中的方法。 (不是 TensorFlow),它封装了 tf.tensor_scatter_nd_add 中描述的行为,但封装在 Numpy 数组而不是张量上。
我遇到了 scipy.ndimage.sum 方法,但无法重现我在下面给出的示例。
您认为合适的方法必须能够重现 TF 文档中提供的 3 级示例:
indices = tf.constant([[0],[2]])
updates = tf.constant([[[5,5,5],[6,6,6],[7,7,7],[8,8,8]],[[5,8]]])
tensor = tf.ones([4,4,4],dtype=tf.int32)
updated = tf.tensor_scatter_nd_add(tensor,indices,updates)
print(updated)
希望有人之前解决过类似的问题并能在这里提供帮助 - 提前致谢!
解决方法
我可以确认以下函数为我捕获了所需的行为:
def scatter_nd_add_numpy(target,indices,updates):
indices = tuple(indices.reshape(-1,indices.shape[-1]).T)
np.add.at(target,updates)
return target
感谢 Remy 在 this stackoverflow thread 上的回答。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。