Tree-Sliced Wasserstein Distance with Nonlinear Projection