embedding

Module Contents

Classes

PositionalEncodingU2

positional encoding can be used in transformer U2

class embedding.PositionalEncodingU2(d_model, max_position=5000, dropout_rate=0.0, scale=False)

Bases: tensorflow.keras.layers.Layer

positional encoding can be used in transformer U2

call(x, offset: int = 0)

call function