diff --git a/docs/feature_extraction.md b/docs/feature_extraction.md
index 3d638d65..86a0e28e 100644
--- a/docs/feature_extraction.md
+++ b/docs/feature_extraction.md
@@ -90,6 +90,7 @@ print(f'Pooled shape: {o.shape}')
 ```
 Output:
 ```text
+Original shape: torch.Size([2, 1000])
 Pooled shape: torch.Size([2, 1024])
 ```