search

coreylowman / dfdx

Deep learning in Rust, with shape checked tensors and neural networks
Other
1.71k stars 99 forks source link

Make tests pass on stable + support Pool2D & Conv{1,2}d #927

Open twitchyliquid64 opened 2 months ago

twitchyliquid64 commented 2 months ago

Simple PR that implements necessary traits manually (instead of relying on generic_const_exprs in nightly), enough to get tests to pass.

Would love to implement common kernel sizes (lmk). Happy to put this behind a feature flag too.

Fixes: #926