Updated docs to specify exactly what layers support outdims

This commit is contained in:
Kyle Daruwalla 2020-02-20 23:38:56 -06:00
parent c37fc3cfa6
commit f5b9cf659c
1 changed files with 11 additions and 0 deletions

View File

@ -225,6 +225,17 @@ This enables a useful extra set of functionality for our `Affine` layer, such as
Flux provides some utility functions to help you generate models in an automated fashion. Flux provides some utility functions to help you generate models in an automated fashion.
`outdims` enables you to calculate the spatial output dimensions of layers like `Conv` when applied to input images of a given size. `outdims` enables you to calculate the spatial output dimensions of layers like `Conv` when applied to input images of a given size.
Currently limited to the following layers:
- `Chain`
- `Dense`
- `Conv`
- `Diagonal`
- `Maxout`
- `ConvTranspose`
- `DepthwiseConv`
- `CrossCor`
- `MaxPool`
- `MeanPool`
```@docs ```@docs
outdims outdims