Merge pull request #1277 from lukasugar/patch-1

Add missing output in Feature extraction docs
pull/1606/head
Ross Wightman 2 years ago committed by GitHub
commit 99af63ca92
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -90,6 +90,7 @@ print(f'Pooled shape: {o.shape}')
``` ```
Output: Output:
```text ```text
Original shape: torch.Size([2, 1000])
Pooled shape: torch.Size([2, 1024]) Pooled shape: torch.Size([2, 1024])
``` ```

Loading…
Cancel
Save