Package deepnetts.cudnn
Class SoftMaxBackwardCuda
java.lang.Object
deepnetts.cudnn.AbstractBackwardCuda
deepnetts.cudnn.SoftMaxBackwardCuda
- All Implemented Interfaces:
BackwardAccelerator
-
Constructor Summary
ConstructorsConstructorDescriptionSoftMaxBackwardCuda
(CudaHandles cudaHandles, SoftmaxOutputLayer layer) SoftMaxBackwardCuda
(CudaHandles cudaHandles, TensorBase deltas, TensorBase weights, TensorBase biases, TensorBase inputs) -
Method Summary
Methods inherited from class deepnetts.cudnn.AbstractBackwardCuda
copyTensorsToDevice
-
Constructor Details
-
SoftMaxBackwardCuda
public SoftMaxBackwardCuda(CudaHandles cudaHandles, TensorBase deltas, TensorBase weights, TensorBase biases, TensorBase inputs) -
SoftMaxBackwardCuda
-
-
Method Details
-
backward
public void backward()Description copied from class:AbstractBackwardCuda
Performs forward pass.- Specified by:
backward
in interfaceBackwardAccelerator
- Specified by:
backward
in classAbstractBackwardCuda
-