ott.geometry.geometry.Geometry.src_mask
ott.geometry.geometry.Geometry.src_mask#
- property Geometry.src_mask: Optional[jax.Array]#
Mask of shape
[num_a,]
to computecost_matrix
statistics.Specifically, it is used when computing:
- Return type
Optional
[Array
]