Merge pull request #1277 from lukasugar/patch-1
Add missing output in Feature extraction docspull/1606/head
commit
99af63ca92
|
@ -90,6 +90,7 @@ print(f'Pooled shape: {o.shape}')
|
|||
```
|
||||
Output:
|
||||
```text
|
||||
Original shape: torch.Size([2, 1000])
|
||||
Pooled shape: torch.Size([2, 1024])
|
||||
```
|
||||
|
||||
|
|
Loading…
Reference in New Issue