Softmax Layer

Softmax layer types and functions

miopenSoftmaxForward

miopenStatus_t miopenSoftmaxForward(miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t xDesc, const void *x, const void *beta, const miopenTensorDescriptor_t yDesc, void *y)

Execute a softmax forward layer.

MIOpen does not support Softmax modes. MIOpen implements the SOFTMAX_MODE_CHANNEL flavor.

Return
miopenStatus_t
Parameters
  • handle: MIOpen handle (input)
  • alpha: Floating point scaling factor, allocated on the host (input)
  • xDesc: Tensor descriptor for data input tensor x (input)
  • x: Data tensor x (input)
  • beta: Floating point shift factor, allocated on the host (input)
  • yDesc: Tensor descriptor for output data tensor y (input)
  • y: Data tensor y (output)

miopenSoftmaxBackward

miopenStatus_t miopenSoftmaxBackward(miopenHandle_t handle, const void *alpha, const miopenTensorDescriptor_t yDesc, const void *y, const miopenTensorDescriptor_t dyDesc, const void *dy, const void *beta, const miopenTensorDescriptor_t dxDesc, void *dx)

Execute a softmax backwards layer.

MIOpen does not support Softmax modes. MIOpen implements the SOFTMAX_MODE_CHANNEL flavor.

Return
miopenStatus_t
Parameters
  • handle: MIOpen handle (input)
  • alpha: Floating point scaling factor, allocated on the host (input)
  • yDesc: Tensor descriptor for input data tensor y (input)
  • y: Data tensor y (input)
  • dyDesc: Tensor descriptor for input data tensor dy (input)
  • dy: Data delta tensor dy (input)
  • beta: Floating point shift factor, allocated on the host (input)
  • dxDesc: Tensor descriptor for data output tensor dx (input)
  • dx: Output data delta tensor dx (output)