fn
scatter_prod_op
→TensorImplPtrint scatter_prod_op(const int & base, const int & indices, const int & src, int dim)Scatter with multiplicative reduction.
Math
Parameters
base, indices, src, dimSame semantics as
scatter_add_op.Returns
TensorImplPtrSame shape and dtype as base.
Notes
Gradients use the product rule:
,
(the latter requires base[i] != 0).