L_Softmax_TensorFlow icon indicating copy to clipboard operation
L_Softmax_TensorFlow copied to clipboard

L_Softmax_TensorFlow

TensorFlow version of L_SoftMax.

Results:

results

I found prelu is quite stable than relu, so I used prelu as paper said.

Reference:

Contribution

This is mainly implemented by py_func, which is quite slow. If anyone have implemented a tf_op in C++ or cuda, pull request is warmly welcome.