ott.initializers.quadratic.initializers.LRQuadraticInitializer#

class ott.initializers.quadratic.initializers.LRQuadraticInitializer(lr_linear_initializer)[source]#

Wrapper that wraps low-rank Sinkhorn initializers.

Parameters:

lr_linear_initializer (LRInitializer) – Low-rank linear initializer.

Methods

Attributes

rank

Rank of the transport matrix factorization.