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

softmax

Daniel_H_
初学者
2,525 次查看

Does MKL support the popular DNN operation called softmax?

I cannot find any suitable function.

0 项奖励
1 解答
Zhen_Z_Intel
员工
2,525 次查看

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

在原帖中查看解决方案

0 项奖励
2 回复数
Zhen_Z_Intel
员工
2,526 次查看

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

0 项奖励
Jingwei_Z_Intel
2,525 次查看
0 项奖励
回复