ott.geometry.geometry.Geometry.src_mask

property Geometry.src_mask: Optional[jax._src.numpy.ndarray.ndarray]

Mask of shape [num_a,] to compute cost_matrix statistics.

Specifically, it is used when computing:

• mean_cost_matrix

• median_cost_matrix

• inv_scale_cost

Return type

Optional[ndarray]