Add missing output

3 years ago · 2b2196028f
parent e4360e6125
commit 2b2196028f
1 changed files with 1 additions and 0 deletions
--- 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])
 ```