Merge pull request #1277 from lukasugar/patch-1

Add missing output in Feature extraction docs
2 years ago · 99af63ca92
parent 45c447fc15 2b2196028f
commit 99af63ca92
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])
 ```