in_to_cos: bool = True
freq_shift: int = 0
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 0x7f15b1bd0520>
name: str = None

)


Parameters 

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


in_channels (int, optional, defaults to 4) —
The number of channels in the input sample.


out_channels