xmartlabs / Bender

Easily craft fast Neural Networks on iOS! Use TensorFlow models. Metal under the hood.
https://xmartlabs.github.io/Bender/
MIT License
1.8k stars 90 forks source link

Add new layers: SpatialNorm, SoftMax, Pooling and Dense #3

Closed mats-claassen closed 7 years ago

mats-claassen commented 7 years ago

Layers must still be tested in a real network.

Trello card

mats-claassen commented 7 years ago

Second commit addresses correctness checks

bryant1410 commented 7 years ago

@mats-claassen I think SoftMax should be just Softmax. At least the term is referred like that everywhere I have seen it.