ott.geometry.low_rank.LRCGeometry.tgt_mask#

property LRCGeometry.tgt_mask: Optional[jax._src.numpy.ndarray.ndarray]#

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

Specifically, it is used when computing:

Return type

Optional[ndarray]