map instead of broadcast in size for Zeros/Ones

[jishnub]

This is a simper operation that unwraps the loop.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.48%. Comparing base (6f61dc3) to head (8e9b1c9).
Report is 6 commits behind head on master.

Additional details and impacted files

@@            Coverage Diff             @@
##           master     #399      +/-   ##
==========================================
- Coverage   99.91%   99.48%   -0.44%     
==========================================
  Files           8        8              
  Lines        1147     1154       +7     
==========================================
+ Hits         1146     1148       +2     
- Misses          1        6       +5     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.