Differentiable Top-k with Optimal Transport

Open in new window