@thi.ng/tensors
Version:
0D/1D/2D/3D/4D tensors with extensible polymorphic operations and customizable storage
16 lines • 640 B
TypeScript
/**
* Componentwise computes step function for given nD tensor and uniform scalar
* threshold `n`. Writes result to `out`. If `out` is null, mutates `a`.
* Multi-method. Also see {@link step} and {@link smoothStep}.
*
* @remarks
* Same logic as GLSL `step()` (but with different order of arguments). If
* `n=0`, the op becomes the Heaviside function:
* https://en.wikipedia.org/wiki/Heaviside_step_function
*
* @param out - output tensor
* @param a - input tensor
* @param n - scalar
*/
export declare const stepN: import("./api.js").MultiTensorOpImpl<import("./api.js").TensorOpTN<number>>;
//# sourceMappingURL=stepn.d.ts.map