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