moscot.base.output.BaseDiscreteSolverOutput.push¶

BaseDiscreteSolverOutput.push(x, scale_by_marginals=False)[source]¶

Push mass through the transport_matrix.

It is equivalent to \(T^T x\) but without instantiating the transport matrix \(T\), if possible.

Parameters:

x (Union[ndarray[tuple[int, ...], dtype[floating]], Array]) – Array of shape [n,] or [n, d] to push.
scale_by_marginals (bool) – Whether to scale by the source marginals a.

Return type:

Returns:

: Array of shape [m,] or [m, d], depending on the shape of x.