Commit Graph

70 Commits

Author SHA1 Message Date
2bfa791336 Use the same default as pytorch for sum. (#164) 2023-07-13 21:32:32 +01:00
50b0946a2d Tensor mutability (#154)
* Working towards tensor mutability.

* Use a ref-cell to provide tensor mutability.
2023-07-13 11:04:40 +01:00
868743b8b9 Expanding a bit the README 2023-07-10 12:51:37 +02:00
5b0ee2e0ba Get cuda to work on pyo3. 2023-07-02 21:04:11 +01:00
fbfe74caab Preliminary pyo3 support for device. 2023-07-02 20:42:55 +01:00
bdb257ceab Add the tensor function. 2023-07-02 20:15:50 +01:00
78871ffe38 Add dtype support. 2023-07-02 20:12:26 +01:00
d38897461b Add to the example. 2023-07-02 07:37:17 +01:00
5b8c6764b0 Add matmul/where_cond. 2023-07-02 07:34:14 +01:00
9a9858bbe0 Expose a couple more ops. 2023-07-02 07:30:00 +01:00
dfe197f791 Handle more input types to create tensors. 2023-07-02 07:19:46 +01:00
4a28dcf828 Rename the method. 2023-07-02 07:08:11 +01:00
c62cb73a7f Support higher order shapes for conversions. 2023-07-02 07:07:22 +01:00
fa58c7643d Add a trait to avoid repeating the dtype matching. 2023-07-02 06:58:10 +01:00
2370b1675d More pyo3. 2023-07-01 22:15:58 +01:00
86df4ad79c Get shape to return a tuple. 2023-07-01 21:34:38 +01:00
fbbde5b02c Add some binary operators. 2023-07-01 21:27:35 +01:00
42d1a52d01 Add two methods. 2023-07-01 20:55:15 +01:00
52db2a6849 Apply rustfmt. 2023-07-01 20:37:28 +01:00
ebb0fedf14 Very simple pyo3 bindings for candle. 2023-07-01 20:36:44 +01:00