FMA

FMA factory Compute array of frequencies over several non-overlapping intervals of a given length

tohubohu.fma.fma(length: int, weights: jax.Array, mapping: Callable[[...], jax.Array]) → Callable[[...], jax.Array][source]

FMA factory (non-overlapping intervals)

Parameters:

  • length (int) – number of intervals

  • weights (Array) – weights to apply

  • mapping (Callable[[Array, *Any], Array]) – state transformation mapping

Return type:

Callable[[Array, *Any], Array]