P = materialize_transport_plan(x, y, f_hat, g_hat, self.eps_f, x2, y2)
Source: torchmatch/transport/samples/kernels/cg_dense.py:467
