Class that computes the results of the forward softmax cross-entropy layer in the batch processing mode.
More...
◆ SoftmaxCrossForwardBatch() [1/2]
Constructs the forward softmax cross-entropy layer by copying input objects of another forward softmax cross-entropy layer
- Parameters
-
context | Context to manage the forward softmax cross-entropy layer |
other | A forward softmax cross-entropy layer to be used as the source to initialize the input objects of the forward softmax cross-entropy layer |
◆ SoftmaxCrossForwardBatch() [2/2]
Constructs the forward softmax cross-entropy layer
- Parameters
-
context | Context to manage the layer |
cls | Data type to use in intermediate computations for the layer, Double.class or Float.class |
method | The layer computation method, SoftmaxCrossMethod |
◆ clone()
Returns the newly allocated forward softmax cross-entropy layer with a copy of input objects of this forward softmax cross-entropy layer
- Parameters
-
context | Context to manage the layer |
- Returns
- The newly allocated forward softmax cross-entropy layer
◆ compute()
Computes the result of the forward softmax cross-entropy layer
- Returns
- Forward softmax cross-entropy layer result
◆ getLayerInput()
Returns the structure that contains input object of the forward layer
- Returns
- Structure that contains input object of the forward layer
◆ getLayerParameter()
Returns the structure that contains parameters of the forward layer
- Returns
- Structure that contains parameters of the forward layer
◆ getLayerResult()
Returns the structure that contains result of the forward layer
- Returns
- Structure that contains result of the forward layer
◆ setResult()
Registers user-allocated memory to store the result of the forward softmax cross-entropy layer
- Parameters
-
result | Structure to store the result of the forward softmax cross-entropy layer |
◆ input
◆ method
Computation method for the layer
◆ parameter
SoftmaxCrossParameters of the layer
The documentation for this class was generated from the following file:
- SoftmaxCrossForwardBatch.java