Skip to main content

//clika-runtime/io.clika.runtime/Ops/maskedScatter

maskedScatter

[common]
fun maskedScatter(input: Tensor, mask: Tensor, source: Tensor): Tensor

maskedScatter(input: Tensor, mask: Tensor, source: Tensor): the masked_scatter operator. A copy of input with the leading count(mask) elements of src (read row-major) written at the positions where mask is true. mask is Bool, broadcastable to input; src must supply at least count(mask) elements.