IntegerLookup

A preprocessing layer that maps integers to (possibly encoded) indices.

Abstract Signature:

IntegerLookup(max_tokens: int, mask_token: int)

PyTorchKerasTensorFlow

PyTorch

API: torch.nn.Embedding

Strategy: Plugin (vocab_lookup)

Keras

API: keras.layers.IntegerLookup

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.keras.layers.IntegerLookup

Strategy: Direct Mapping

Official Docs ↗