ott.geometry.low_rank.LRCGeometry.tgt_mask
ott.geometry.low_rank.LRCGeometry.tgt_mask#
- property LRCGeometry.tgt_mask: Optional[jax.Array]#
Mask of shape
[num_b,]
to computecost_matrix
statistics.Specifically, it is used when computing:
- Return type
Optional
[Array
]