public class GaussianNoise extends Object implements IDropout
| Modifier | Constructor and Description |
|---|---|
|
GaussianNoise(double stddev) |
protected |
GaussianNoise(double stddev,
ISchedule stddevSchedule) |
|
GaussianNoise(ISchedule stddevSchedule) |
| Modifier and Type | Method and Description |
|---|---|
INDArray |
applyDropout(INDArray inputActivations,
INDArray output,
int iteration,
int epoch,
LayerWorkspaceMgr workspaceMgr) |
INDArray |
backprop(INDArray gradAtOutput,
INDArray gradAtInput,
int iteration,
int epoch)
Perform backprop.
|
void |
clear()
Clear the internal state (for example, dropout mask) if any is present
|
IDropout |
clone() |
public GaussianNoise(double stddev)
stddev - Standard deviation for the mean 0 Gaussian noisepublic GaussianNoise(ISchedule stddevSchedule)
stddevSchedule - Schedule for standard deviation for the mean 0 Gaussian noiseprotected GaussianNoise(double stddev,
ISchedule stddevSchedule)
public INDArray applyDropout(INDArray inputActivations, INDArray output, int iteration, int epoch, LayerWorkspaceMgr workspaceMgr)
applyDropout in interface IDropoutinputActivations - Input activations arrayoutput - The result array (same as inputArray for in-place ops) for the post-dropout activationsiteration - Current iteration numberepoch - Current epoch numberworkspaceMgr - Workspace manager, if any storage is required (use ArrayType.INPUT)public INDArray backprop(INDArray gradAtOutput, INDArray gradAtInput, int iteration, int epoch)
IDropoutbackprop in interface IDropoutgradAtOutput - Gradients at the output of the dropout op - i.e., dL/dOutgradAtInput - Gradients at the input of the dropout op - i.e., dL/dIn. Use the same array as gradAtOutput
to apply the backprop gradient in-placeiteration - Current iterationepoch - Current epochpublic void clear()
IDropoutCopyright © 2021. All rights reserved.