![]() |
Fourier Transform layer.
tfm.nlp.layers.FourierTransformLayer(
use_fft: bool = False, name: str = 'fourier_transform', **kwargs
)
Applies 2D Fourier Transform over final two dimensions of query
inputs -
typically the sequence and hidden dimensions.
Methods
call
call(
query: tf.Tensor, value: tf.Tensor, **kwargs
) -> tf.Tensor
Applies layer to query
.
Args | |
---|---|
query
|
Batch of input embeddings, typically of shape |
value
|
Unused. Included to match attention layer API. |
**kwargs
|
Optional arguments to catch unused attention keyword arguments. |
Returns | |
---|---|
Real part of discrete Fourier Transform of query inputs with shape
|