oneAPI Deep Neural Network Library (oneDNN)
Performance library for Deep Learning
1.96.0

This C++ API example demonstrates how to create and execute a Softmax primitive in forward training propagation mode.

Key optimizations included in this example:

  • In-place primitive execution;
  • Softmax along axis 1 (C) for 2D tensors.