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

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.