ott.geometry.low_rank.LRKGeometry.tgt_mask

property LRKGeometry.tgt_mask: Array | None

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

Specifically, it is used when computing:

• mean_cost_matrix

• median_cost_matrix

• inv_scale_cost