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

Add some tensor creation functions to the pyo3 bindings. (#326)

Browse Source
This commit is contained in:
Laurent Mazare
2023-08-06 06:50:33 +01:00
committed by GitHub
parent b278834267
commit 88bd3b604a
3 changed files with 77 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-pyo3/test.py
View File

@ -13,3 +13,6 @@ t = t.reshape([2, 4])
print(t.matmul(t.t()))
print(t.to_dtype("u8"))
t = candle.randn((5, 3))
print(t)