Reducer

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

Base class for reducers.

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

Extended by:

• opennmt.layers.ConcatReducer

• opennmt.layers.JoinReducer

• opennmt.layers.MultiplyReducer

• opennmt.layers.SumReducer

zip_and_reduce(x, y)

Zips the x with y structures together and reduces all elements. If the structures are nested, they will be flattened first.

Parameters

• x – The first structure.

• y – The second structure.

Returns

The same structure as x and y where each element from x is reduced with the correspond element from y.

Raises

ValueError – if the two structures are not the same.

call(inputs, sequence_length=None)

Reduces all input elements.

Parameters

• inputs – A list of tf.Tensor.

• sequence_length – The length of each input, if reducing sequences.

Returns

If sequence_length is set, a tuple (reduced_input, reduced_length), otherwise a reduced tf.Tensor only.

abstract reduce(inputs)

See opennmt.layers.Reducer.call().

abstract reduce_sequence(inputs, sequence_lengths)

See opennmt.layers.Reducer.call().