![]() |
Hartley Transform layer.
tfm.nlp.layers.HartleyTransformLayer(
use_fft: bool = False, name: str = 'hartley_transform', **kwargs
)
Applies 2D Hartley 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 Hartley Transform of query inputs with shape
|