scatter_nd
该OP根据 index
,将 updates
添加到一个新的张量中,从而得到输出的Tensor。这个操作与 scatter_nd_add
类似,除了形状为 shape
的张量是通过零初始化的。相应地, 等价于 scatter_nd_add(fluid.layers.zeros(shape, updates.dtype), index, updates)
。如果 index
有重复元素,则将累积相应的更新,因此,由于数值近似问题,索引中重复元素的顺序不同可能会导致不同的输出结果。具体的计算方法可以参见 scatter_nd_add
。该OP是 gather_nd
的反函数。
参数:
返回:数据类型与 updates
相同,形状是 shape
的Tensor。
返回类型:Tensor