25 lines
760 B
Python
25 lines
760 B
Python
"""
|
|
scatter
|
|
"""
|
|
|
|
from keras_core import backend
|
|
from keras_core.api_export import keras_core_export
|
|
from keras_core.backend import KerasTensor
|
|
from keras_core.backend import any_symbolic_tensors
|
|
from keras_core.operations.operation import Operation
|
|
|
|
|
|
class Scatter(Operation):
|
|
def call(self, indices, values, shape):
|
|
return backend.core.scatter(indices, values, shape)
|
|
|
|
def compute_output_spec(self, indices, values, shape):
|
|
return KerasTensor(shape, dtype=values.dtype)
|
|
|
|
|
|
@keras_core_export("keras_core.operations.scatter")
|
|
def scatter(indices, values, shape):
|
|
if any_symbolic_tensors((indices, values, shape)):
|
|
return Scatter().symbolic_call(indices, values, shape)
|
|
return backend.core.scatter(indices, values, shape)
|