Package deepnetts.cudnn
Class FullyConnectedForwardCuda
java.lang.Object
deepnetts.cudnn.AbstractForwardCuda
deepnetts.cudnn.FullyConnectedForwardCuda
- All Implemented Interfaces:
ForwardAccelerator
-
Constructor Summary
ConstructorsConstructorDescriptionFullyConnectedForwardCuda(CudaHandles cudaHandles, FullyConnectedLayer layer) FullyConnectedForwardCuda(CudaHandles cudaHandles, TensorBase outputs, TensorBase weights, TensorBase biases, TensorBase inputs) -
Method Summary
Methods inherited from class deepnetts.cudnn.AbstractForwardCuda
copyTensorsToDevice
-
Constructor Details
-
FullyConnectedForwardCuda
public FullyConnectedForwardCuda(CudaHandles cudaHandles, TensorBase outputs, TensorBase weights, TensorBase biases, TensorBase inputs) - Parameters:
cudaHandles- cuda library context handlesoutputs- 1D tensor with outputs for this layerweights- 2D tensor with weightsbiases- 1D tensor with biasesinputs- 1D tensor with inputs from previous layer maybe change order of tensors? i, w, o | mada out = W * X where * is dot product make sure that outputs contains biases and the beta is 1 Ovo izbaci imas novi konstruktor ispod koji je zamena ikoji mora da ima referncu na layer KOristim je samo za testiranje
-
FullyConnectedForwardCuda
-
-
Method Details
-
forward
public void forward()Description copied from class:AbstractForwardCudaPerforms forward pass.- Specified by:
forwardin interfaceForwardAccelerator- Specified by:
forwardin classAbstractForwardCuda
-