cuda.js/lib/utils/elementwise.d.ts

CUDA bindings for Node.js

import { GpuArray } from '../gpu-array';
/**
 * Elementwise operations on GPU arrays
 */
export declare class ElementwiseKernel {
    private kernel;
    private operation;
    constructor(operation: string, functionName?: string);
    apply(a: GpuArray, b: GpuArray | number, out?: GpuArray): GpuArray;
    free(): void;
}
export declare const elementwise: {
    add: (a: GpuArray, b: GpuArray | number, out?: GpuArray) => GpuArray;
    subtract: (a: GpuArray, b: GpuArray | number, out?: GpuArray) => GpuArray;
    multiply: (a: GpuArray, b: GpuArray | number, out?: GpuArray) => GpuArray;
    divide: (a: GpuArray, b: GpuArray | number, out?: GpuArray) => GpuArray;
    power: (a: GpuArray, b: number, out?: GpuArray) => GpuArray;
    exp: (a: GpuArray, out?: GpuArray) => GpuArray;
    log: (a: GpuArray, out?: GpuArray) => GpuArray;
    sqrt: (a: GpuArray, out?: GpuArray) => GpuArray;
};
//# sourceMappingURL=elementwise.d.ts.map