Skip to main content

//clika-runtime/io.clika.runtime/Ops/indexFill

indexFill

[common]
fun indexFill(input: Tensor, dim: Long, indices: Tensor, value: Tensor): Tensor

indexFill(input: Tensor, dim: Long, indices: Tensor, value: Tensor): the index_fill operator. A copy of input with the rows at indices along dim set to value: out[.., indices[i], ..] = value. indices is 1-D Int32/Int64; value is a literal or a 0-D Tensor.

[common]
fun indexFill(input: Tensor, dim: Long, indices: Tensor, value: Double): Tensor

indexFill(input: Tensor, dim: Long, indices: Tensor, value: Double): the number form of index_fill, value as a scalar.