src.model.deeplearn.layer.expansion_layer

Classes

ExpansionLayer(*args, **kwargs)

class src.model.deeplearn.layer.expansion_layer.ExpansionLayer(*args, **kwargs)

Author:

Alberto M. Esmoris Pena

Layer wrapping the expand_dims method from keras.ops.

Variables:

axis (int) – The index for the new axis.

__init__(axis, **kwargs)

See Layer and Layer.__init__().

call(x, training=False, mask=False)

Compute the actual expansion.

Parameters:

x – The input tensor

Returns:

The expanded tensor.

get_config()

Return necessary data to deserialize the layer.

classmethod from_config(config)

Use given config data to deserialize the layer