Skip to main content

//clika-runtime/io.clika.runtime/Ops/maskedFill

maskedFill

[common]
fun maskedFill(input: Tensor, mask: Tensor, value: Tensor): Tensor

maskedFill(input: Tensor, mask: Tensor, value: Tensor): the masked_fill operator. Replace the elements of input where mask is true with value. mask is Bool and broadcasts to input's shape; value is a literal or a 0-D Tensor.

[common]
fun maskedFill(input: Tensor, mask: Tensor, value: Double): Tensor

maskedFill(input: Tensor, mask: Tensor, value: Double): the number form of masked_fill, value as a scalar.