huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add to the example.

Browse Source
This commit is contained in:
laurent
2023-07-02 07:37:17 +01:00
parent 5b8c6764b0
commit d38897461b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
candle-pyo3/test.py
View File

@ -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()))