TorchEx
TorchEx copied to clipboard
Collection of PyTorch customized operators.
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