Intel® oneAPI Math Kernel Library
Ask questions and share information with other developers who use Intel® Math Kernel Library.

softmax

Daniel_H_
Beginner
323 Views

Does MKL support the popular DNN operation called softmax?

I cannot find any suitable function.

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

Jingwei_Z_Intel
Employee
323 Views
Reply