search

transcranial / keras-js

Run Keras models in the browser, with GPU support using WebGL
https://transcranial.github.io/keras-js
MIT License
4.96k stars 503 forks source link

Add another dimension to softmax #118

Open linusmartensson opened 6 years ago

linusmartensson commented 6 years ago

The softmax activation in the model I received was in three dimensions, so I added support for that layer shape to the CPU version of the softmax activation by simply extending another dimension to the library implementation. Wouldn't mind if someone takes a look at a 3D-GPU version? :1234: :)

seranus commented 6 years ago

thanks was just look for this

seranus commented 6 years ago

I'm getting errors with you code changes, why are you using _ndarrayOps.default instead of ops?