kd.nn.convert_to_fourier_features#

kauldron.modules.convert_to_fourier_features( inputs: jaxtyping.Float[Array, '... D'] | jaxtyping.Float[ndarray, '... D'], basis_degree: int, precision: None | str | jax._src.lax.lax.Precision | tuple[str, str] | tuple[jax._src.lax.lax.Precision, jax._src.lax.lax.Precision] | jax._src.lax.lax.DotAlgorithm | jax._src.lax.lax.DotAlgorithmPreset = Precision.HIGHEST, ) → jaxtyping.Float[Array, '... d'] | jaxtyping.Float[ndarray, '... d'][source]: Convert inputs to Fourier features, e.g. for positional encoding.