ttnn.index_fill

ttnn.index_fill() → None

index_fill(input: Tensor, dim: uint32, index: Tensor, value: int or float, memory_config: MemoryConfig) -> Tensor Create or fill a tensor with the given value, with the specified memory_config. This operation only supports ROW_MAJOR_LAYOUT for now. :param * input: The tensor that we will operate on :param * dim: The dimension that we need to fill the value along. :param * index: The index that we need to fill the value in. :param * value: The value which will be used to fill the output tensor :param * memory_config: The memory configuration for the output tensor.