org.nd4j.linalg.api.ops.custom.BarnesHutGains 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.api.ops.custom;

import lombok.val;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.api.ops.DynamicCustomOp;

/**
 * This op calculates gains - data used internally by Barnes-Hut-TSNE algorithm.
 *
 * @author [email protected]
 */
public class BarnesHutGains extends DynamicCustomOp {

    public BarnesHutGains(){ }

    public BarnesHutGains(INDArray output, INDArray input, INDArray gradx, INDArray epsilon) {

        inputArguments.add(input);
        inputArguments.add(gradx);
        inputArguments.add(epsilon);

        outputArguments.add(output);
    }

    @Override
    public String opName() {
        return "barnes_gains";
    }
}