Interface for calculating the final step of the Softmax Layer where each logit value is multiplied by the inverse of the sum of the logits.
More...
#include <NESoftmaxLayerKernel.h>
Interface for calculating the final step of the Softmax Layer where each logit value is multiplied by the inverse of the sum of the logits.
Definition at line 94 of file NESoftmaxLayerKernel.h.
Prevent instances of this class from being copied (As this class contains pointers)
Allow instances of this class to be moved.
Set the input and output tensors.
- Parameters
-
[in] | input | Source tensor. Data types supported: F32. |
[in] | sum | Sum tensor. The number of dimensions should be dim(input)-1. Data types supported: same as input . |
[out] | output | Destination tensor. Data types supported: same as input . |
Prevent instances of this class from being copied (As this class contains pointers)
Allow instances of this class to be moved.
void run |
( |
const Window & |
window | ) |
|
|
overridevirtual |
Execute the kernel on the passed window.
- Warning
- If is_parallelisable() returns false then the passed window must be equal to window()
- Note
- The window has to be a region within the window returned by the window() method
-
The width of the window has to be a multiple of num_elems_processed_per_iteration().
- Parameters
-
[in] | window | Region on which to execute the kernel. (Must be a region of the window returned by window()) |
Implements ICPPKernel.
The documentation for this class was generated from the following file: