org.nd4j.autodiff.samediff.ops.SDOps Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of nd4j-api Show documentation

There is a newer version: 1.0.0-M2.1

package org.nd4j.autodiff.samediff.ops;

import org.nd4j.autodiff.functions.DifferentialFunctionFactory;
import org.nd4j.autodiff.samediff.SDVariable;
import org.nd4j.autodiff.samediff.SameDiff;
import org.nd4j.linalg.api.buffer.DataType;

import java.util.Arrays;

/**
 * Abstract class for defining categories of operations - such as {@link SDMath} that is available via {@code SameDiff.math()}
 *
 * @author Alex Black
 */
public abstract class SDOps {

    protected final SameDiff sd;

    public SDOps(SameDiff sameDiff) {
        this.sd = sameDiff;
    }

    protected DifferentialFunctionFactory f() {
        return sd.f();
    }

    protected SDVariable updateVariableNameAndReference(SDVariable varToUpdate, String newVarName) {
        return sd.updateVariableNameAndReference(varToUpdate, newVarName);
    }
}