From d38897461b5e2947bf6b552a7453a078dc7a9021 Mon Sep 17 00:00:00 2001 From: laurent Date: Sun, 2 Jul 2023 07:37:17 +0100 Subject: [PATCH] Add to the example. --- candle-pyo3/test.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/candle-pyo3/test.py b/candle-pyo3/test.py index aad5e8ae..d63f752b 100644 --- a/candle-pyo3/test.py +++ b/candle-pyo3/test.py @@ -5,7 +5,8 @@ print(t) print("shape", t.shape, t.rank) print(t + t) -t = candle.Tensor([3, 1, 4, 1, 5, 9, 2, 6]) +t = candle.Tensor([3.0, 1, 4, 1, 5, 9, 2, 6]) print(t) print(t+t) -print(t.reshape([2, 4])) +t = t.reshape([2, 4]) +print(t.matmul(t.t()))