I started exploring the mojo programming language and now I am trying to figure out how to properly use the Tensor module. I can't find how to statically declare the values inside a tensor.
For the moment, this is what I do to fill a tensor with values:
let dim1 = 2
let dim2 = 3
var matrice1 = Tensor[DType.float32](dim1, dim2)
for i in range(dim1):
for j in range(dim2):
matrice1[Index(i,j)] = 1
But I am looking for a way to do something like this:
var matrice1 = Tensor[DType.float32](dim1, dim2)
matrice1 = [[1, 2, 3],[1, 2, 3]]
You can initialize a Tensor from a literal like this:
let dim1 = 2
let dim2 = 3
let tensor = Tensor[DType.float32](
TensorShape(dim1, dim2),
1, 2, 3,
1, 2, 3
)
print(tensor)