on: bool = False
dtype: dtype = <class 'jax.numpy.float32'>
flip_sin_to_cos: bool = True
freq_shift: int = 0
use_memory_efficient_attention: bool = False
parent: typing.Union[typing.Type[flax.linen.module.Module], typing.Type[flax.core.scope.Scope], typing.Type[flax.linen.module._Sentinel], NoneType] = <flax.linen.module._Sentinel object at 0x7ff2943c0550>
name: str = None

)


Parameters 

sample_size (int, optional) —
The size of the input sa