Files
candle/candle-pyo3/_additional_typing
Lukas Kreussel c05c0a8213 PyO3: Add equal and __richcmp__ to candle.Tensor (#1099)
* add `equal` to tensor

* add `__richcmp__` support  for tensors and scalars

* typo

* more typos

* Add `abs` + `candle.testing`

* remove duplicated `broadcast_shape_binary_op`

* `candle.i16` => `candle.i64`

* `tensor.nelements` -> `tensor.nelement`

* Cleanup `abs`
2023-10-30 15:17:28 +00:00
..

This python module contains external typehinting for certain candle classes. This is only necessary for magic methodes e.g. __add__ as their text signature cant be set via pyo3.

The classes in this module will be parsed by the stub.py script and interleafed with the signatures of the actual pyo3 candle.candle module.