ott.geometry.graph.Graph.tgt_mask
ott.geometry.graph.Graph.tgt_mask#
- property Graph.tgt_mask: Optional[jax.Array]#
Mask of shape
[num_b,]to computecost_matrixstatistics.Specifically, it is used when computing:
- Return type
Optional[Array]
Mask of shape [num_b,] to compute cost_matrix statistics.
Specifically, it is used when computing:
Optional[Array]