How to statically declare a Tensor?

Question

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]]

ephemer · Accepted Answer · 2023-12-26 12:41:48Z

4

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)

answered Dec 26, 2023 at 12:41

ephemer

1,3249 silver badges24 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

ephemer Over a year ago

Note that Tensor is now considered deprecated in Mojo and will be removed – at least from the standard library – soon (see youtu.be/uIG9q9foIw0?si=Me9Ll6BczscHh6fa&t=1947)

Collectives™ on Stack Overflow

How to statically declare a Tensor?

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related