ott.geometry.low_rank.LRKGeometry.src_mask

property LRKGeometry.src_mask: Array | None

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