Intel® oneAPI Math Kernel Library
Ask questions and share information with other developers who use Intel® Math Kernel Library.
Announcements
Intel Customer Support will be observing the Martin Luther King holiday on Monday, Jan. 17, and will return on Tues. Jan. 18.
For the latest information on Intel’s response to the Log4j/Log4Shell vulnerability, please see Intel-SA-00646

softmax

Daniel_H_
Beginner
214 Views

Does MKL support the popular DNN operation called softmax?

I cannot find any suitable function.

0 Kudos
1 Solution
Zhen_Z_Intel
Employee
214 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
215 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
214 Views
Reply