TorchEx icon indicating copy to clipboard operation
TorchEx copied to clipboard

Collection of PyTorch customized operators.

Results 4 TorchEx issues
Sort by recently updated
recently updated
newest added

Why can't I use the torchex.connected_components to get the same result as scipy.parse.csgraph.connected_components? Sometimes it's the same, but sometimes it's different, and the initial estimate is 50-50。 the code is...

SST model relies on function 'IngroupIndicesFunction', which needs ingroup_indices op. But it is unable to export to torchscript. How can I fix it. Error message: Could not export Python function...

https://github.com/Abyssaledge/TorchEx/blob/e5ff7fae27b5b6df74e6204dac45554ba85117a2/torchex/src/dynamic_point_pool/dynamic_point_pool_kernel.cu#LL297C41-L297C41 I'm confused about this variable and this code. Can you explain it? Thank you~

Detailed changes: - added pyproject.toml - modified setup.py