ott.geometry.graph.Graph.src_mask#

property Graph.src_mask: Optional[jax._src.basearray.Array]#

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

Specifically, it is used when computing:

Return type

Optional[Array]