fn main() {
    let t = tensor(1.0, 2.0, 3.0, 4.0)
    println(t)
    println(tensor_sum(t))

    let embed_table = t_rand(100, 32)
    let indices = tensor(5.0, 10.0, 15.0)
    let embedded = embedding_lookup(embed_table, indices)
    println(tensor_shape(embedded))
}
