Embedding layers in TensorFlow - Practice Problems & Coding Challenges

Challenge - 5 Problems

❓ Predict Output

intermediate

Output shape of embedding layer

What is the shape of the output tensor after passing input indices through this embedding layer?

TensorFlow

import tensorflow as tf
embedding = tf.keras.layers.Embedding(input_dim=10, output_dim=4)
input_indices = tf.constant([[1, 2, 3], [4, 5, 6]])
output = embedding(input_indices)
print(output.shape)

A (3, 2, 4)

B (2, 3, 4)

C (2, 4, 3)

D (3, 4)

💡 Hint

Remember the output shape is (batch_size, sequence_length, embedding_dim).

The input has shape (2,3) representing 2 samples each with 3 indices. The embedding layer converts each index to a vector of size 4, so output shape is (2,3,4).

❓ Model Choice

intermediate

Choosing embedding input dimension

You have a vocabulary of 5000 unique words. Which input_dim should you use for the embedding layer?

A input_dim=5000

B input_dim=5001

C input_dim=10000

D input_dim=4999

💡 Hint

input_dim should be at least the max index + 1.

The input_dim parameter must be set to the size of the vocabulary, i.e. maximum integer index +1. For 5000 unique words indexed 0 to 4999, input_dim=5000.

❓ Hyperparameter

advanced

Effect of embedding dimension size

What is the main effect of increasing the embedding output_dim size in a model?

A It reduces the model's capacity and may underfit.

B It changes the input vocabulary size.

C It increases the model's capacity and may overfit if too large.

D It decreases the training time significantly.

💡 Hint

Think about how embedding size affects model complexity.

Larger embedding dimensions allow the model to learn richer representations but increase parameters, which can lead to overfitting if not enough data is available.

🔧 Debug

advanced

Identify the error in embedding usage

What error will this code raise when run?

TensorFlow

import tensorflow as tf
embedding = tf.keras.layers.Embedding(input_dim=5, output_dim=3)
input_indices = tf.constant([[1, 2, 5]])
output = embedding(input_indices)

A IndexError: index 5 is out of bounds for axis 0 with size 5

B ValueError: input_dim must be greater than output_dim

C TypeError: input_indices must be float type

D No error, runs successfully

💡 Hint

Check if input indices are within the allowed range.

The input_dim=5 means valid indices are 0 to 4. Index 5 is out of range and causes an IndexError.

🧠 Conceptual

expert

Why use embedding layers instead of one-hot encoding?

Which reason best explains why embedding layers are preferred over one-hot encoding for representing words in neural networks?

A One-hot encoding can represent words with fewer dimensions than embeddings.

B One-hot encoding is faster to train and uses less memory.

C Embedding layers require no training and are fixed representations.

D Embedding layers reduce dimensionality and capture semantic relationships between words.

💡 Hint

Think about how embeddings represent meaning compared to one-hot vectors.

Embedding layers learn dense vector representations that capture word meanings and relationships, unlike sparse one-hot vectors which treat words as unrelated.

Practice

1. What is the main purpose of an Embedding layer in TensorFlow?

easy

A. To reduce the dimensionality of numerical data using PCA

B. To normalize input data between 0 and 1

C. To perform convolution operations on images

D. To convert categorical data or words into dense vectors of numbers

Solution

1. Step 1: Understand the role of embedding layers

   Embedding layers transform categorical or text data into vectors that capture relationships.

2. Step 2: Compare with other options

   Normalization, convolution, and PCA are different processes unrelated to embedding layers.

3. Final Answer:

   To convert categorical data or words into dense vectors of numbers -> Option D

4. Quick Check:

   Embedding layer = vector conversion [OK]

Hint: Embedding layers turn categories into number vectors [OK]

Common Mistakes:

• Confusing embedding with normalization
• Thinking embedding is for images
• Mixing embedding with dimensionality reduction

2. Which of the following is the correct way to create an embedding layer in TensorFlow for 1000 words with 64-dimensional vectors?

easy

A. tf.keras.layers.Embedding(input_dim=1000, output_dim=64)

B. tf.keras.layers.Embedding(input_dim=64, output_dim=1000)

C. tf.keras.layers.Dense(1000, 64)

D. tf.keras.layers.Embedding(1000, 64, input_length=1000)

Solution

1. Step 1: Identify correct parameters for Embedding

   Embedding requires input_dim as vocabulary size and output_dim as vector size.

2. Step 2: Check each option

   tf.keras.layers.Embedding(input_dim=1000, output_dim=64) correctly uses input_dim=1000 and output_dim=64. tf.keras.layers.Embedding(input_dim=64, output_dim=1000) swaps them. tf.keras.layers.Dense(1000, 64) uses Dense layer incorrectly. tf.keras.layers.Embedding(1000, 64, input_length=1000) wrongly sets input_length to 1000, which is usually sequence length, not vocabulary size.

3. Final Answer:

   tf.keras.layers.Embedding(input_dim=1000, output_dim=64) -> Option A

4. Quick Check:

   input_dim = vocab size, output_dim = vector size [OK]

Hint: input_dim = vocab size, output_dim = embedding size [OK]

Common Mistakes:

• Swapping input_dim and output_dim
• Using Dense layer instead of Embedding
• Confusing input_length with input_dim

3. What will be the output shape of the embedding layer for input shape (batch_size=32, sequence_length=10) if the embedding layer is defined as Embedding(input_dim=5000, output_dim=128)?

medium

A. (32, 10, 5000)

B. (32, 128, 10)

C. (32, 10, 128)

D. (10, 32, 128)

Solution

1. Step 1: Understand input and output shapes of Embedding

   Embedding converts each integer in the input sequence to a vector of length output_dim. Input shape is (batch_size, sequence_length).

2. Step 2: Calculate output shape

   For input shape (32, 10), output shape becomes (32, 10, 128) because each of the 10 tokens is replaced by a 128-dimensional vector.

3. Final Answer:

   (32, 10, 128) -> Option C

4. Quick Check:

   Output shape = (batch, sequence, embedding size) [OK]

Hint: Output shape adds embedding size as last dimension [OK]

Common Mistakes:

• Confusing input_dim with output shape
• Swapping batch and sequence dimensions
• Ignoring embedding vector size in output

4. You wrote this code but get an error:

embedding = tf.keras.layers.Embedding(input_dim=1000, output_dim=64)
input_data = tf.constant([[1, 2, 3], [4, 5, 1001]])
output = embedding(input_data)

What is the cause of the error?

medium

A. The input contains an index (1001) outside the vocabulary size

B. The output_dim is too large for the input

C. Embedding layer requires input shape to be 1D

D. The input data type must be float, not int

Solution

1. Step 1: Check input indices against input_dim

   Embedding input indices must be in range [0, input_dim-1]. Here, input_dim=1000, so max index is 999.

2. Step 2: Identify invalid index

   Input contains 1001, which is out of range and causes an error.

3. Final Answer:

   The input contains an index (1001) outside the vocabulary size -> Option A

4. Quick Check:

   Input indices must be less than input_dim [OK]

Hint: Check input indices are less than input_dim [OK]

Common Mistakes:

• Ignoring out-of-range indices
• Thinking input must be 1D
• Confusing data types for embedding input

5. You want to use an embedding layer for a text classification model. Your vocabulary size is 5000, and you want embeddings of size 50. You also want to limit the input sequences to length 20. Which of the following code snippets correctly creates the embedding layer and input shape for this model?

hard

A. embedding = tf.keras.layers.Embedding(input_dim=5000, output_dim=20, input_length=50)

B. embedding = tf.keras.layers.Embedding(input_dim=5000, output_dim=50, input_length=20)

C. embedding = tf.keras.layers.Embedding(input_dim=20, output_dim=50, input_length=5000)

D. embedding = tf.keras.layers.Embedding(input_dim=50, output_dim=5000, input_length=20)

Solution

1. Step 1: Match parameters to problem requirements

   Vocabulary size is 5000, so input_dim=5000. Embedding size is 50, so output_dim=50. Input sequences length is 20, so input_length=20.

2. Step 2: Validate each option

   embedding = tf.keras.layers.Embedding(input_dim=5000, output_dim=50, input_length=20) matches all requirements correctly. Options A, B, and D have incorrect parameter assignments.

3. Final Answer:

   embedding = tf.keras.layers.Embedding(input_dim=5000, output_dim=50, input_length=20) -> Option B

4. Quick Check:

   input_dim = vocab, output_dim = embed size, input_length = seq length [OK]

Hint: Match input_dim, output_dim, input_length to vocab, embed size, seq length [OK]

Common Mistakes:

• Swapping input_dim and input_length
• Confusing output_dim with input_length
• Using wrong values for embedding size