Mojo 1.0 Beta

def vector_add(
    a: TileTensor[float_dtype, type_of(layout), element_size=1, ...],
    b: TileTensor[float_dtype, type_of(layout), element_size=1, ...],
    result: TileTensor[
        mut=True, float_dtype, type_of(layout), element_size=1, ...
    ],
):
    var i = global_idx.x
    if i < layout.size():
        result[i] = a[i] + b[i]