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
..
README.md Extend `stub.py` to accept external typehinting (#1102) 2023-10-17 11:07:26 +01:00
__init__.py PyO3: Add `equal` and `__richcmp__` to `candle.Tensor` (#1099) 2023-10-30 15:17:28 +00:00

README.md

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.