deepof.model_utils.create_masks

deepof.model_utils.create_masks(inp: Tensor)

Given an input sequence, it creates all necessary masks to pass it through the transformer architecture.

This includes encoder and decoder padding masks, and a look-ahead mask

Parameters:

inp (tf.Tensor) – input sequence to create the masks for.