It seems that a similar option in Flux.jl is @autosize. But we still need to manually pass in the first input layer dimension.
It seems that a similar option in Flux.jl is @autosize. But we still need to manually pass in the first input layer dimension.