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._src.basearray.Array) –

  • y (jax._src.basearray.Array) –

  • static_b (bool) –

  • src_mask (Optional[jax._src.basearray.Array]) –

  • tgt_mask (Optional[jax._src.basearray.Array]) –

  • kwargs (Any) –