Community
cancel
Showing results for 
Search instead for 
Did you mean: 
Highlighted
Beginner
38 Views

Does MKL support the popular DNN operation called softmax?

I cannot find any suitable function.

0 Kudos

Accepted Solutions
Highlighted
Employee
38 Views

Hi Daniel,

I am afraid MKL does not provide softmax, only ReLU for activation. Actually, MKL only provides few functions for NN, mainly probably for convolution. You may could refer to use MKL-DNN (mkldnn_softmax_forward_desc_init) which can be optimized with MKL VM & BLAS functions. Thanks.

Best regards,
Fiona

View solution in original post

0 Kudos
2 Replies
Highlighted
Employee
39 Views

Hi Daniel,

I am afraid MKL does not provide softmax, only ReLU for activation. Actually, MKL only provides few functions for NN, mainly probably for convolution. You may could refer to use MKL-DNN (mkldnn_softmax_forward_desc_init) which can be optimized with MKL VM & BLAS functions. Thanks.

Best regards,
Fiona

View solution in original post

0 Kudos
Highlighted
38 Views
0 Kudos