![]() 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> |
||
---|---|---|
.. | ||
cuda | ||
layers | ||
data.jl | ||
onehot.jl | ||
optimise.jl | ||
runtests.jl | ||
utils.jl |