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

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

Key optimizations included in this example:

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