
org.deeplearning4j.spark.impl.common.SumAccum Maven / Gradle / Ivy
package org.deeplearning4j.spark.impl.common;
import org.apache.spark.Accumulator;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.factory.Nd4j;
import scala.Option;
/**
* @author Adam Gibson
*/
public class SumAccum extends Accumulator {
public SumAccum(int length) {
super(Nd4j.zeros(length), new ParamAccumulator());
}
@Override
public Option name() {
return Option.empty();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy