Add missing output

This commit is contained in:
lukasugar 2022-05-26 19:23:23 +02:00 committed by GitHub
parent e4360e6125
commit 2b2196028f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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