neureka.backend.main.implementations.elementwise.CPUBiElementWiseMultiplication Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of neureka Show documentation

A platform independent tensor library written in Java.

The newest version!

package neureka.backend.main.implementations.elementwise;

import neureka.backend.main.implementations.fun.api.CPUBiFun;

public class CPUBiElementWiseMultiplication extends CPUBiElementWise
{
    @Override
    protected CPUBiFun _getFun() {
        return new CPUBiFun() {
            @Override public double invoke(double a, double b) { return a * b; }
            @Override public float invoke(float a, float b) { return a * b; }
            @Override public int invoke(int a, int b) { return a * b; }
        };
    }

    @Override
    protected CPUBiFun _getDeriveAt0() {
        return new CPUBiFun() {
            @Override public double invoke(double a, double b) { return b; }
            @Override public float invoke(float a, float b) { return b; }
            @Override public int invoke(int a, int b) { return b; }
        };
    }

    @Override
    protected CPUBiFun _getDeriveAt1() {
        return new CPUBiFun() {
            @Override public double invoke(double a, double b) { return a; }
            @Override public float invoke(float a, float b) { return a; }
            @Override public int invoke(int a, int b) { return a; }
        };
    }
}