如何静态声明Tensor？

Question

我开始探索 mojo 编程语言，现在我正在尝试弄清楚如何正确使用 Tensor 模块。我找不到如何静态声明张量内的值。

目前，这就是我用值填充张量的方法：

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

但我正在寻找一种方法来做这样的事情：

var matrice1 = Tensor[DType.float32](dim1, dim2)
matrice1 = [[1, 2, 3],[1, 2, 3]]

Answer 1

您可以从这样的文字初始化张量：

let dim1 = 2
let dim2 = 3

let tensor = Tensor[DType.float32](
    TensorShape(dim1, dim2),
    1, 2, 3,
    1, 2, 3
)

print(tensor)