ott.geometry.low_rank.LRCGeometry.tgt_mask