candle/candle-core
Thomas Santerre c5626b8271
Add support for "sign" on tensors (#2012)
* add the sign unary operator

* remove uneeded import

* remove uneeded import

* undo formatting

* undo formatting

* remove unnecessary redefintion

* allow gradient to flow through for sign and round

* fix cpu ops to ensure that negzero and positive zero are handled properly

* clippy fixes

* Properly avoid gradient tracking.

* Use a branchless version.

---------

Co-authored-by: laurent <laurent.mazare@gmail.com>
2024-04-04 22:32:47 +02:00
..
benches Add support for conv_transpose2d on Metal backend (#1903) 2024-03-21 18:08:45 +01:00
examples Move the tensor-tools binary in a separate crate. (#1969) 2024-03-30 15:49:37 +01:00
src Add support for "sign" on tensors (#2012) 2024-04-04 22:32:47 +02:00
tests Add support for "sign" on tensors (#2012) 2024-04-04 22:32:47 +02:00
Cargo.toml feat(bf16): add cast support + tests for cast + bin ops (#1524) 2024-01-11 15:49:13 +01:00
LICENSE Refactor the hierarchy. 2023-06-27 11:57:27 +02:00
README.md Refactor the hierarchy. 2023-06-27 11:57:27 +02:00

README.md

candle

Minimalist ML framework for Rust