public interface LayerModel
extends scala.Serializable
Modifier and Type | Method and Description |
---|---|
void |
computePrevDelta(breeze.linalg.DenseMatrix<Object> delta,
breeze.linalg.DenseMatrix<Object> output,
breeze.linalg.DenseMatrix<Object> prevDelta)
Computes the delta for back propagation.
|
void |
eval(breeze.linalg.DenseMatrix<Object> data,
breeze.linalg.DenseMatrix<Object> output)
Evaluates the data (process the data through the layer).
|
void |
grad(breeze.linalg.DenseMatrix<Object> delta,
breeze.linalg.DenseMatrix<Object> input,
breeze.linalg.DenseVector<Object> cumGrad)
Computes the gradient.
|
breeze.linalg.DenseVector<Object> |
weights() |
void computePrevDelta(breeze.linalg.DenseMatrix<Object> delta, breeze.linalg.DenseMatrix<Object> output, breeze.linalg.DenseMatrix<Object> prevDelta)
delta
- delta of this layeroutput
- output of this layerprevDelta
- the previous delta (modified in place)void eval(breeze.linalg.DenseMatrix<Object> data, breeze.linalg.DenseMatrix<Object> output)
data
- dataoutput
- output (modified in place)void grad(breeze.linalg.DenseMatrix<Object> delta, breeze.linalg.DenseMatrix<Object> input, breeze.linalg.DenseVector<Object> cumGrad)
delta
- delta for this layerinput
- input datacumGrad
- cumulative gradient (modified in place)breeze.linalg.DenseVector<Object> weights()