PositionEncoder

class opennmt.layers.PositionEncoder(*args, **kwargs)

Base class for position encoders.

Inherits from: keras.src.engine.base_layer.Layer

Extended by:

• opennmt.layers.PositionEmbedder

• opennmt.layers.SinusoidalPositionEncoder

__init__(reducer=None, **kwargs)

Initializes the position encoder.

Parameters

• reducer – A opennmt.layers.Reducer to merge inputs and position encodings. Defaults to opennmt.layers.SumReducer.

• **kwargs – Additional layer keyword arguments.

call(inputs, position=None)

Add position encodings to inputs.

Parameters

• inputs – The inputs to encode.

• position – The single position to encode, to use when this layer is called step by step.

Returns

A tf.Tensor whose shape depends on the configured reducer.