55616afc11
960: Added utility function outdims to compute output dimensions of a layer r=dhairyagandhi96 a=darsnack Based on Slack chatter, I added a utility function, `outdims`, that computes the output dimensions for given input dimensions. Example ```julia layer = Conv((3, 3), 3 => 16) outdims(layer, (10, 10)) # returns (8, 8) ``` Co-authored-by: Kyle Daruwalla <daruwalla@wisc.edu> |
||
|---|---|---|
| .github | ||
| docs | ||
| paper | ||
| src | ||
| test | ||
| .gitattributes | ||
| .gitignore | ||
| .gitlab-ci.yml | ||
| .travis.yml | ||
| CITATION.bib | ||
| LICENSE.md | ||
| Manifest.toml | ||
| NEWS.md | ||
| Project.toml | ||
| README.md | ||
| bors.toml | ||
README.md
Flux is an elegant approach to machine learning. It's a 100% pure-Julia stack, and provides lightweight abstractions on top of Julia's native GPU and AD support. Flux makes the easy things easy while remaining fully hackable.
] add Flux
See the documentation or the model zoo for examples.
If you use Flux in your research, please cite our work.