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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of samediff

There is a newer version: 0.9.1

package org.nd4j.autodiff.functions;

import lombok.NoArgsConstructor;
import org.nd4j.autodiff.Field;
import org.nd4j.autodiff.samediff.SDGraph;

@NoArgsConstructor
public abstract class AbstractBinaryFunction> extends DifferentialFunction {

    protected DifferentialFunction m_x1;
    protected DifferentialFunction m_x2;

    public AbstractBinaryFunction(SDGraph graph,
                                  DifferentialFunction i_v1,
                                  DifferentialFunction i_v2) {
        super(graph,new Object[] {i_v2});
        if (i_v1 != null && i_v2 != null) {
            m_x1 = i_v1;
            m_x2 = i_v2;
            this.graph = graph;

            addEdges(graph,i_v1,i_v2,functionName());
        } else {
            throw new IllegalArgumentException("Input not null variables.");
        }
    }

    public AbstractBinaryFunction(SDGraph graph) {
        this.graph = graph;
    }


    @Override
    public DifferentialFunction[] args() {
        return new DifferentialFunction[] {larg(),rarg()};
    }

    @Override
    public DifferentialFunction arg() {
        return larg();
    }

    public DifferentialFunction larg() {
        if(m_x1 == this)
            return m_x1.dup();
        return m_x1;
    }


    public DifferentialFunction rarg() {
        if(m_x2 == this)
            return m_x2.dup();
        return m_x2;
    }



    @Override
    public DifferentialFunction dup() {
        try {
            return getClass().getConstructor(graph.getClass(),larg().getClass(),rarg().getClass()).newInstance(graph,larg(),rarg());
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
}