org.nd4j.linalg.ops.ArrayOps 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.linalg.ops;

import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.util.ReflectionUtil;

import java.lang.reflect.Constructor;

/**
 * Builder for element wise operations
 *
 * @author Adam Gibson
 */
public class ArrayOps {

    private INDArray from,scalar;
    private Class clazz;
    private Object[] extraArgs;


    /**
     * Extra arguments for a constructor
     * @param extraArgs the extra arguments for a constructor
     * @return
     */
    public ArrayOps extraArgs(Object[] extraArgs) {
        this.extraArgs = extraArgs;
        return this;
    }

    /**
     * The operation to perform
     * @param clazz the class of the operation to perform
     * @return builder pattern
     */
    public ArrayOps op(Class clazz) {
        this.clazz = clazz;
        return this;
    }


    public ArrayOps from(INDArray from) {
        this.from = from;
        return this;
    }


    public ArrayOps scalar(INDArray scalar) {
        this.scalar = scalar;
        return this;
    }


    public ElementWiseOp build() {
        try {
            ElementWiseOp op;
            if(extraArgs == null)
                op = clazz.newInstance();
            else {
                Constructor c = clazz.getConstructor(ReflectionUtil.classesFor(extraArgs));
                op = (ElementWiseOp) c.newInstance(extraArgs);
            }
            BaseElementWiseOp op2 = (BaseElementWiseOp) op;
            op2.from = from;
            op2.scalarValue = scalar;
            return op;
        }catch (Exception e) {
            throw new RuntimeException(e);

        }
    }

}