oneAPI Deep Neural Network Library (oneDNN)
Performance library for Deep Learning
1.96.0
dnnl::logsoftmax_forward::desc Struct Reference

Descriptor for a logsoftmax forward propagation primitive. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::logsoftmax_forward::desc:

Public Member Functions

 desc ()=default
 Default constructor. Produces an empty object.
 
 desc (prop_kind aprop_kind, const memory::desc &data_desc, int logsoftmax_axis)
 Constructs a descriptor for a logsoftmax forward propagation primitive. More...
 

Detailed Description

Descriptor for a logsoftmax forward propagation primitive.

Examples:
logsoftmax.cpp.

Constructor & Destructor Documentation

◆ desc()

dnnl::logsoftmax_forward::desc::desc ( prop_kind  aprop_kind,
const memory::desc data_desc,
int  logsoftmax_axis 
)
inline

Constructs a descriptor for a logsoftmax forward propagation primitive.

Parameters
aprop_kindPropagation kind. Possible values are dnnl::prop_kind::forward_training, and dnnl::prop_kind::forward_inference.
data_descSource and destination memory descriptor.
logsoftmax_axisAxis over which softmax is computed.

The documentation for this struct was generated from the following file: