Practice Now
Python
Error Handling
Data Structures
Debugging
Programming

ValueError The two structures don't have the same number of elements

Data Structures & Algorithms practice on Codemia

Step through 300 algorithm problems with animated visualisers that show the data structure changing as the code runs.

Practice algorithms
Data StructuresPythonDebugging & Errors

Introduction

The ValueError: The two structures don't have the same number of elements error occurs in TensorFlow and Keras when two nested structures (tuples, lists, or dictionaries) that are expected to match element-by-element have different lengths or shapes. This commonly happens when a model's output signature does not match the loss functions, the dataset returns a different number of elements than the model expects, or when using tf.nest utilities on mismatched structures. The fix is to ensure both structures have identical nesting and element counts.

Common Cause 1: Model Output vs Loss Function Mismatch

python
1import tensorflow as tf
2
3# Model returns 2 outputs
4inputs = tf.keras.Input(shape=(10,))
5x = tf.keras.layers.Dense(64, activation='relu')(inputs)
6output1 = tf.keras.layers.Dense(1, name='regression')(x)
7output2 = tf.keras.layers.Dense(5, activation='softmax', name='classification')(x)
8
9model = tf.keras.Model(inputs=inputs, outputs=[output1, output2])
10
11# WRONG — only 1 loss for 2 outputs
12model.compile(optimizer='adam', loss='mse')
13# ValueError: The two structures don't have the same number of elements
14
15# CORRECT — one loss per output
16model.compile(
17    optimizer='adam',
18    loss={
19        'regression': 'mse',
20        'classification': 'categorical_crossentropy'
21    }
22)
23
24# Or as a list
25model.compile(optimizer='adam', loss=['mse', 'categorical_crossentropy'])

Common Cause 2: Dataset Shape Mismatch

python
1# Model expects (features, labels) but dataset returns (features, label1, label2)
2def wrong_generator():
3    for i in range(100):
4        features = tf.random.normal((10,))
5        label1 = tf.constant(1.0)
6        label2 = tf.constant([0, 1, 0, 0, 0])
7        yield features, label1, label2  # 3 elements
8
9dataset = tf.data.Dataset.from_generator(
10    wrong_generator,
11    output_signature=(
12        tf.TensorSpec(shape=(10,), dtype=tf.float32),
13        tf.TensorSpec(shape=(), dtype=tf.float32),
14        tf.TensorSpec(shape=(5,), dtype=tf.int32),
15    )
16)
17
18# WRONG — model.fit expects (features, labels) where labels matches outputs
19model.fit(dataset)
20# ValueError: structure mismatch
21
22# CORRECT — pack labels into a dict matching output names
23def correct_generator():
24    for i in range(100):
25        features = tf.random.normal((10,))
26        labels = {
27            'regression': tf.constant(1.0),
28            'classification': tf.constant([0, 1, 0, 0, 0], dtype=tf.float32)
29        }
30        yield features, labels
31
32dataset = tf.data.Dataset.from_generator(
33    correct_generator,
34    output_signature=(
35        tf.TensorSpec(shape=(10,), dtype=tf.float32),
36        {
37            'regression': tf.TensorSpec(shape=(), dtype=tf.float32),
38            'classification': tf.TensorSpec(shape=(5,), dtype=tf.float32),
39        }
40    )
41)
42
43model.fit(dataset, epochs=5)

Common Cause 3: tf.nest Operations

python
1import tensorflow as tf
2
3structure1 = (1, 2, 3)
4structure2 = (4, 5)
5
6# WRONG — different lengths
7tf.nest.map_structure(lambda a, b: a + b, structure1, structure2)
8# ValueError: The two structures don't have the same number of elements.
9# First structure: (3 elements), second structure: (2 elements)
10
11# CORRECT — same length
12structure2 = (4, 5, 6)
13result = tf.nest.map_structure(lambda a, b: a + b, structure1, structure2)
14print(result)  # (5, 7, 9)

Common Cause 4: Custom Training Loop with Multiple Outputs

python
1# WRONG — unpacking mismatch
2@tf.function
3def train_step(inputs, labels):
4    with tf.GradientTape() as tape:
5        predictions = model(inputs)  # Returns [output1, output2]
6        # Trying to compute loss with wrong structure
7        loss = tf.keras.losses.mse(labels, predictions)  # labels is single tensor
8    return loss
9
10# CORRECT — handle multiple outputs explicitly
11@tf.function
12def train_step(inputs, labels):
13    with tf.GradientTape() as tape:
14        pred_regression, pred_classification = model(inputs)
15        loss1 = tf.keras.losses.mse(labels['regression'], pred_regression)
16        loss2 = tf.keras.losses.categorical_crossentropy(
17            labels['classification'], pred_classification
18        )
19        total_loss = loss1 + loss2
20    gradients = tape.gradient(total_loss, model.trainable_variables)
21    optimizer.apply_gradients(zip(gradients, model.trainable_variables))
22    return total_loss

Debugging the Error

python
1import tensorflow as tf
2
3# Inspect structure shapes
4structure1 = model.output
5structure2 = losses
6
7print("Model outputs:", tf.nest.map_structure(lambda x: x.shape, structure1))
8print("Losses:", tf.nest.map_structure(lambda x: type(x).__name__, structure2))
9
10# Count elements
11print("Output count:", len(tf.nest.flatten(structure1)))
12print("Loss count:", len(tf.nest.flatten(structure2)))
13
14# Assert structures match
15try:
16    tf.nest.assert_same_structure(structure1, structure2)
17    print("Structures match")
18except ValueError as e:
19    print(f"Mismatch: {e}")

Common Pitfalls

  • Single loss for multi-output model: Passing one loss function to model.compile() when the model has multiple outputs causes this error. Provide a list or dictionary of losses matching the number and names of outputs.
  • Dataset returning wrong tuple structure: model.fit(dataset) expects each element to be (inputs, labels) or (inputs, labels, sample_weights). If the dataset yields a different number of elements or a mismatched nested structure, the structures will not align.
  • Mixing list and tuple nesting: TensorFlow's tf.nest treats lists and tuples as different structure types. (1, [2, 3]) and (1, (2, 3)) are different structures. Ensure consistent use of lists or tuples throughout.
  • Dictionary key mismatch: When using dictionary-based labels, the keys must match the model's output layer names exactly. {'output_1': ...} does not match a layer named 'regression'. Check model.output_names to verify the expected keys.
  • RNN returning sequences with wrong label shape: An RNN with return_sequences=True outputs shape (batch, timesteps, features), but labels may be shape (batch, features). The structural mismatch between 3D predictions and 2D labels triggers this error. Reshape labels or adjust return_sequences.

Summary

  • This error means two nested structures have different element counts or nesting shapes
  • For multi-output models, provide one loss per output in compile() as a list or dictionary
  • Dataset generators must return (inputs, labels) where labels matches the model's output structure
  • Use tf.nest.assert_same_structure() to debug mismatches before training
  • Check model.output_names to ensure dictionary keys in labels and losses match output layer names
Related reading
Course
Intermediate
27 lessons
15 hours
DSA Fundamentals

Master algorithmic patterns and data structures through hands-on LeetCode-style problems - from arrays and hashing to dynamic programming and advanced graphs.

View the course
Track what you have practised

A free account saves your progress, solutions and study plan across every problem on Codemia.

Data Structures & Algorithms practice on Codemia

Step through 300 algorithm problems with animated visualisers that show the data structure changing as the code runs.

Practice algorithms
Codemia
© 2026 Codemia
Resources
Legal
Social
All Rights Reserved.