ott.geometry.low_rank.LRCGeometry.tgt_mask

property LRCGeometry.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