gensbi.flow_matching.utils package

Subpackages

• gensbi.flow_matching.utils.manifolds_not_implemented package
  • Submodules
  • gensbi.flow_matching.utils.manifolds_not_implemented.manifold module
    • Euclidean
      • Euclidean.expmap()
      • Euclidean.logmap()
      • Euclidean.proju()
      • Euclidean.projx()
    • Manifold
      • Manifold.expmap()
      • Manifold.logmap()
      • Manifold.proju()
      • Manifold.projx()
  • gensbi.flow_matching.utils.manifolds_not_implemented.sphere module
  • gensbi.flow_matching.utils.manifolds_not_implemented.torus module
  • gensbi.flow_matching.utils.manifolds_not_implemented.utils module
  • Module contents

Submodules

gensbi.flow_matching.utils.utils module

gensbi.flow_matching.utils.utils.expand_tensor_like(input_array: Array, expand_to: Array) → Array

input_array is a 1d vector of length equal to the batch size of expand_to, expand input_array to have the same shape as expand_to along all remaining dimensions.

Parameters:

• input_array (Array) – (batch_size,).

• expand_to (Array) – (batch_size, …).

Returns:

(batch_size, …).

Return type:

Array

gensbi.flow_matching.utils.utils.unsqueeze_to_match(source: Array, target: Array, how: str = 'suffix') → Array

Unsqueeze the source array to match the dimensionality of the target array.

Parameters:

• source (Array) – The source array to be unsqueezed.

• target (Array) – The target array to match the dimensionality of.

• how (str, optional) – Whether to unsqueeze the source array at the beginning (“prefix”) or end (“suffix”). Defaults to “suffix”.

Returns:

The unsqueezed source array.

Return type:

Array

Module contents

gensbi.flow_matching.utils.expand_tensor_like(input_array: Array, expand_to: Array) → Array

input_array is a 1d vector of length equal to the batch size of expand_to, expand input_array to have the same shape as expand_to along all remaining dimensions.

Parameters:

• input_array (Array) – (batch_size,).

• expand_to (Array) – (batch_size, …).

Returns:

(batch_size, …).

Return type:

Array

gensbi.flow_matching.utils.unsqueeze_to_match(source: Array, target: Array, how: str = 'suffix') → Array

Unsqueeze the source array to match the dimensionality of the target array.

Parameters:

• source (Array) – The source array to be unsqueezed.

• target (Array) – The target array to match the dimensionality of.

• how (str, optional) – Whether to unsqueeze the source array at the beginning (“prefix”) or end (“suffix”). Defaults to “suffix”.

Returns:

The unsqueezed source array.

Return type:

Array