View source on GitHub |
Returns a tensor of shape shape
where indices
are set to values
.
tf.keras.ops.scatter(
indices, values, shape
)
At a high level, this operation does zeros[indices] = updates
and
returns the output. It is equivalent to:
zeros = keras.ops.zeros(shape)
output = keras.ops.scatter_update(zeros, indices, values)
Args | |
---|---|
indices
|
A tensor or list/tuple specifying
indices for the values in values .
|
values
|
A tensor, the values to be set at indices .
|
shape
|
Shape of the output tensor. |
Example:
indices = [[0, 1], [1, 1]]
values = np.array([1., 1.])
keras.ops.scatter(indices, values, shape=(2, 2))
array([[0., 1.],
[0., 1.]])