org.deeplearning4j.spark.impl.common

Class ParamAccumulator

java.lang.Object

org.deeplearning4j.spark.impl.common.ParamAccumulator

All Implemented Interfaces:

Serializable, org.apache.spark.AccumulableParam<org.nd4j.linalg.api.ndarray.INDArray,org.nd4j.linalg.api.ndarray.INDArray>, org.apache.spark.AccumulatorParam<org.nd4j.linalg.api.ndarray.INDArray>

public class ParamAccumulator
extends Object
implements org.apache.spark.AccumulatorParam<org.nd4j.linalg.api.ndarray.INDArray>

Accumulator for addition of parameters

Author:

Adam Gibson

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from interface org.apache.spark.AccumulatorParam

org.apache.spark.AccumulatorParam.DoubleAccumulatorParam$, org.apache.spark.AccumulatorParam.FloatAccumulatorParam$, org.apache.spark.AccumulatorParam.IntAccumulatorParam$, org.apache.spark.AccumulatorParam.LongAccumulatorParam$

Constructor Summary

Constructors

Constructor and Description
ParamAccumulator()

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	addAccumulator(org.nd4j.linalg.api.ndarray.INDArray indArray, org.nd4j.linalg.api.ndarray.INDArray t1)
org.nd4j.linalg.api.ndarray.INDArray	addInPlace(org.nd4j.linalg.api.ndarray.INDArray indArray, org.nd4j.linalg.api.ndarray.INDArray r1)
org.nd4j.linalg.api.ndarray.INDArray	zero(org.nd4j.linalg.api.ndarray.INDArray indArray)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ParamAccumulator

public ParamAccumulator()

Method Detail

addAccumulator

public org.nd4j.linalg.api.ndarray.INDArray addAccumulator(org.nd4j.linalg.api.ndarray.INDArray indArray,
                                                           org.nd4j.linalg.api.ndarray.INDArray t1)

Specified by:

addAccumulator in interface org.apache.spark.AccumulableParam<org.nd4j.linalg.api.ndarray.INDArray,org.nd4j.linalg.api.ndarray.INDArray>

Specified by:

addAccumulator in interface org.apache.spark.AccumulatorParam<org.nd4j.linalg.api.ndarray.INDArray>

addInPlace

public org.nd4j.linalg.api.ndarray.INDArray addInPlace(org.nd4j.linalg.api.ndarray.INDArray indArray,
                                                       org.nd4j.linalg.api.ndarray.INDArray r1)

Specified by:

addInPlace in interface org.apache.spark.AccumulableParam<org.nd4j.linalg.api.ndarray.INDArray,org.nd4j.linalg.api.ndarray.INDArray>

zero

public org.nd4j.linalg.api.ndarray.INDArray zero(org.nd4j.linalg.api.ndarray.INDArray indArray)

Specified by:

zero in interface org.apache.spark.AccumulableParam<org.nd4j.linalg.api.ndarray.INDArray,org.nd4j.linalg.api.ndarray.INDArray>