ott.geometry.pointcloud.PointCloud.prepare_divergences#

classmethod PointCloud.prepare_divergences(x, y, static_b=False, src_mask=None, tgt_mask=None, **kwargs)[source]#

Instantiate the geometries used for a divergence computation.

Return type

Tuple[PointCloud, ...]

Parameters

  • x (jax.Array) –

  • y (jax.Array) –

  • static_b (bool) –

  • src_mask (Optional[jax.Array]) –

  • tgt_mask (Optional[jax.Array]) –

  • kwargs (Any) –