diff options
| author | laurent <laurent.mazare@gmail.com> | 2023-07-02 07:19:46 +0100 |
|---|---|---|
| committer | laurent <laurent.mazare@gmail.com> | 2023-07-02 07:19:46 +0100 |
| commit | dfe197f791b9462c83d7ec9cc141886c868628a7 (patch) | |
| tree | 0ea137c48b5b876bc6c37d75ba9fe1803766d547 /candle-pyo3/test.py | |
| parent | 4a28dcf828ba8393a8167218a0df18fa4ad1f37c (diff) | |
| download | candle-dfe197f791b9462c83d7ec9cc141886c868628a7.tar.gz candle-dfe197f791b9462c83d7ec9cc141886c868628a7.tar.bz2 candle-dfe197f791b9462c83d7ec9cc141886c868628a7.zip | |
Handle more input types to create tensors.
Diffstat (limited to 'candle-pyo3/test.py')
| -rw-r--r-- | candle-pyo3/test.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/candle-pyo3/test.py b/candle-pyo3/test.py index 0db3fab9..aad5e8ae 100644 --- a/candle-pyo3/test.py +++ b/candle-pyo3/test.py @@ -4,3 +4,8 @@ t = candle.Tensor(42.0) print(t) print("shape", t.shape, t.rank) print(t + t) + +t = candle.Tensor([3, 1, 4, 1, 5, 9, 2, 6]) +print(t) +print(t+t) +print(t.reshape([2, 4])) |