oneflow.scatter_nd¶

oneflow.scatter_nd(index, update, shape)¶

This operator inserts the elements in update according to the index and create a new Tensor.

Parameters

index – The indices of update. Its type should be flow.int.
update – The update Tensor.
shape (Sequence[int]) – The constant tensor shape, the constant tensor elements are all zero.

For example:

>>> import oneflow as flow
>>> import numpy as np
>>> index = flow.tensor(np.array([[1], [6], [4]]), dtype=flow.int)
>>> update = flow.tensor(np.array([10.2, 5.1, 12.7]), dtype=flow.float)
>>> out = flow.scatter_nd(index, update, [8])
>>> out
tensor([ 0.0000, 10.2000,  0.0000,  0.0000, 12.7000,  0.0000,  5.1000,  0.0000],
       dtype=oneflow.float32)