public class OutputLayerGradient extends Object implements Serializable
Constructor and Description |
---|
OutputLayerGradient(org.nd4j.linalg.api.ndarray.INDArray wGradient,
org.nd4j.linalg.api.ndarray.INDArray bGradient) |
Modifier and Type | Method and Description |
---|---|
void |
add(OutputLayerGradient gradient)
Sums this gradient with the given one
|
void |
div(int num)
Divies the gradient by the given number (used in averaging)
|
boolean |
equals(Object obj) |
org.nd4j.linalg.api.ndarray.INDArray |
getbGradient() |
org.nd4j.linalg.api.ndarray.INDArray |
getwGradient() |
int |
hashCode() |
void |
setbGradient(org.nd4j.linalg.api.ndarray.INDArray bGradient) |
void |
setwGradient(org.nd4j.linalg.api.ndarray.INDArray wGradient) |
public OutputLayerGradient(org.nd4j.linalg.api.ndarray.INDArray wGradient, org.nd4j.linalg.api.ndarray.INDArray bGradient)
public void div(int num)
num
- the number to divide bypublic void add(OutputLayerGradient gradient)
gradient
- the gradient to addpublic org.nd4j.linalg.api.ndarray.INDArray getwGradient()
public void setwGradient(org.nd4j.linalg.api.ndarray.INDArray wGradient)
public org.nd4j.linalg.api.ndarray.INDArray getbGradient()
public void setbGradient(org.nd4j.linalg.api.ndarray.INDArray bGradient)
Copyright © 2014. All rights reserved.