21.08
|
Functions | |
PermutationVector | get_permutation_vector_from_softmax_axis (size_t axis) |
Given a softmax axis, this function returns the permutation vector required to put the axis to the front. More... | |
PermutationVector get_permutation_vector_from_softmax_axis | ( | size_t | axis | ) |
Given a softmax axis, this function returns the permutation vector required to put the axis to the front.
Axis selects the dimension on which softmax is performed. E.g. For input of shape 4x5x6 and axis=1, softmax will be applied to 4x6=24 vectors of size 5. Interally softmax kernels is always performed on the first dimension (front dimension), therefore permutation is required to put the dimension specified by axis
to the first dimension.
[in] | axis | Axis on which to perform softmax. Supported: 1, 2, 3 (0 implies no permutation needed) |
Definition at line 30 of file SoftmaxHelpers.cpp.
References ARM_COMPUTE_ERROR, and arm_compute::utils::cast::U.
Referenced by ClSoftmax::configure(), CpuSoftmaxGeneric< IS_LOG >::configure(), ClSoftmax::validate(), and CpuSoftmaxGeneric< IS_LOG >::validate().