relax input size

[YichengDWu]

Closes #54

[codecov]

Codecov Report

Merging #57 (fc8ab99) into main (dbe0834) will increase coverage by 0.68%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##             main      #57      +/-   ##
==========================================
+ Coverage   85.89%   86.58%   +0.68%     
==========================================
  Files           2        2              
  Lines         156      164       +8     
==========================================
+ Hits          134      142       +8     
  Misses         22       22              

Impacted Files	Coverage Δ
src/layers.jl	85.71% <100.00%> (+0.29%)	⬆️
src/utils.jl	94.11% <100.00%> (+2.45%)	⬆️

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 7c96745...fc8ab99. Read the comment docs.