scalable-pytorch-sinkhorn
scalable-pytorch-sinkhorn copied to clipboard
Feature request: Batch support
Currently, running the method on a batch of point cloud pairs requires iterating over the batch. a parallelized batched version would be very useful. Thanks for a great repo.
+1!
@sitzikbs were you able to solve this problem?
+1 for the feature request.