Adjust sliced wasserstein

aidos-lab / pytorch-topological

A topological machine learning framework based on PyTorch

https://pytorch-topological.readthedocs.io/

BSD 3-Clause "New" or "Revised" License

142 stars 18 forks source link

Adjust sliced wasserstein #36

Closed Clancy97 closed 5 months ago

Clancy97 commented 5 months ago

The original for loop considerably slowed the running time down as it iterated over all entries of a Persistence Diagram. For large diagrams, this is not desirable, therefore I replaced it with a broadcasting operation.

Pseudomanifold commented 5 months ago

Thanks, looks great!