Intel® oneAPI Math Kernel Library
Ask questions and share information with other developers who use Intel® Math Kernel Library.
Announcements
Welcome to the Intel Community. If you get an answer you like, please mark it as an Accepted Solution to help others. Thank you!

softmax

Daniel_H_
Beginner
196 Views

Does MKL support the popular DNN operation called softmax?

I cannot find any suitable function.

0 Kudos
1 Solution
Zhen_Z_Intel
Employee
196 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

2 Replies
Zhen_Z_Intel
Employee
197 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

Jingwei_Z_Intel
Employee
196 Views
Reply