Pooling by Sliced-Wasserstein Embedding - Naderializadeh, Navid; Comer, Joseph F; Andrews, Reed; Hoffmann, Heiko; Kolouri, Soheil (Advances in Neural Information Processing Systems, 2021)