moscot.backends.ott.GraphOTTOutput.push

GraphOTTOutput.push(x, scale_by_marginals=False)

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[Any, 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:

Union[ndarray[Any, dtype[floating]], Array]

Returns:

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