langml.plm.layers

Module Contents

Classes

TokenEmbedding

EmbeddingMatching

Masked

Generate output mask based on the given mask.
class langml.plm.layers.TokenEmbedding[source]

Bases: tensorflow.keras.layers.Embedding

static get_custom_objects() → dict[source]
compute_mask(self, inputs: langml.tensor_typing.Tensors, mask: Optional[langml.tensor_typing.Tensors] = None) → List[Union[langml.tensor_typing.Tensors, None]][source]
call(self, inputs: langml.tensor_typing.Tensors) → List[langml.tensor_typing.Tensors][source]
compute_output_shape(self, input_shape: langml.tensor_typing.Tensors) → List[langml.tensor_typing.Tensors][source]
class langml.plm.layers.EmbeddingMatching(initializer: langml.tensor_typing.Initializer = 'zeros', regularizer: Optional[langml.tensor_typing.Regularizer] = None, constraint: Optional[langml.tensor_typing.Constraint] = None, use_bias: bool = True, use_softmax: bool = True, **kwargs)[source]

Bases: tensorflow.keras.layers.Layer

get_config(self) → dict[source]
build(self, input_shape: langml.tensor_typing.Tensors)[source]
compute_mask(self, inputs: langml.tensor_typing.Tensors, mask: Optional[langml.tensor_typing.Tensors] = None) → langml.tensor_typing.Tensors[source]
call(self, inputs: langml.tensor_typing.Tensors, mask: Optional[langml.tensor_typing.Tensors] = None, **kwargs) → langml.tensor_typing.Tensors[source]
static get_custom_objects() → dict[source]
compute_output_shape(self, input_shape: langml.tensor_typing.Tensors) → langml.tensor_typing.Tensors[source]
class langml.plm.layers.Masked(return_masked: bool = False, **kwargs)[source]

Bases: tensorflow.keras.layers.Layer

Generate output mask based on the given mask. https://arxiv.org/pdf/1810.04805.pdf

static get_custom_objects() → dict[source]
get_config(self) → dict[source]
compute_mask(self, inputs: langml.tensor_typing.Tensors, mask: Optional[langml.tensor_typing.Tensors] = None) → Union[List[Union[langml.tensor_typing.Tensors, None]], langml.tensor_typing.Tensors][source]
call(self, inputs: langml.tensor_typing.Tensors, mask: Optional[langml.tensor_typing.Tensors] = None, **kwargs) → langml.tensor_typing.Tensors[source]
compute_output_shape(self, input_shape: langml.tensor_typing.Tensors) → Union[List[langml.tensor_typing.Tensors], langml.tensor_typing.Tensors][source]