org.nd4j.autodiff.functions.Sum Maven / Gradle / Ivy

Go to download
package org.nd4j.autodiff.functions;

import java.util.List;

import org.nd4j.autodiff.ArrayField;
import org.nd4j.autodiff.Field;
import org.nd4j.autodiff.samediff.SDGraph;
import org.nd4j.autodiff.samediff.SameDiff;
import org.nd4j.linalg.api.ops.impl.transforms.arithmetic.AddOp;


public class Sum> extends AbstractBinaryFunction {

    public Sum(SameDiff sameDiff, DifferentialFunction i_v1, DifferentialFunction i_v2) {
        super(sameDiff,i_v1, i_v2);
    }

    @Override
    public ArrayField doGetValue() {
        return larg().getValue(true).add(rarg().getValue(true));
    }

    @Override
    public double getReal() {
        return larg().getReal() + rarg().getReal();
    }

    @Override
    public DifferentialFunction diff(DifferentialFunction i_v1) {
        return (larg() == rarg()) ? larg().diff(i_v1).mul(2L) // Field is
                                                              // commutative
                                                              // with respect to
                                                              // addition.
                : larg().diff(i_v1).add(rarg().diff(i_v1));
    }

    @Override
    public String toString() {
        return "(" + larg().toString() + "+" + rarg().toString() + ")";
    }

    @Override
    public String doGetFormula(List> variables) {
        return "(" + larg().doGetFormula(variables) + "+" + rarg().doGetFormula(variables) + ")";
    }

    @Override
    public String functionName() {
        return new AddOp().name();
    }
}