deepbox/ndarray

Math & Reductions

Elementwise arithmetic, transcendental math, logical kernels, sorting helpers, and dense tensor reductions.

Math kernel

corrcoef

export declare function corrcoef(x: Tensor): Tensor;

Compute the Pearson correlation coefficient matrix.

cov

export declare function cov(x: Tensor, ddof?: number): Tensor;

Compute the covariance matrix.

tensordot

export declare function tensordot(a: Tensor, b: Tensor, axes?: number | [number[], number[]]): Tensor;

Tensor dot product along specified axes.

dot

export declare function dot(a: Tensor, b: Tensor): Tensor;

dot is exported by deepbox/ndarray.

abs

export declare function abs(t: Tensor): Tensor;

Element-wise absolute value.

acos

export declare function acos(t: Tensor): Tensor;

Element-wise inverse cosine.

acosh

export declare function acosh(t: Tensor): Tensor;

Element-wise inverse hyperbolic cosine.

add

export declare function add(a: Tensor, b: Tensor): Tensor;

Element-wise addition.

addScalar

export declare function addScalar(t: Tensor, s: number): Tensor;

Add a scalar value to all elements of a tensor.

all

export declare function all(t: Tensor, axis?: number | number[], keepdims?: boolean): Tensor;

Test whether all array elements evaluate to True (non-zero).

allclose

export declare function allclose(a: Tensor, b: Tensor, rtol?: number, atol?: number): boolean;

Test whether all corresponding elements are close within tolerance.

any

export declare function any(t: Tensor, axis?: number | number[], keepdims?: boolean): Tensor;

Test whether any array element evaluates to True (non-zero).

argsort

export declare function argsort(t: Tensor, axis?: Axis | undefined, descending?: boolean): Tensor;

Return indices that would sort the tensor along a given axis.

arrayEqual

export declare function arrayEqual(a: Tensor, b: Tensor): boolean;

Test for exact array equality (shape, dtype, and all values).

asin

export declare function asin(t: Tensor): Tensor;

Element-wise inverse sine.

asinh

export declare function asinh(t: Tensor): Tensor;

Element-wise inverse hyperbolic sine.

atan

export declare function atan(t: Tensor): Tensor;

Element-wise inverse tangent.

atan2

export declare function atan2(y: Tensor, x: Tensor): Tensor;

Element-wise arctangent of y/x with correct quadrant.

atanh

export declare function atanh(t: Tensor): Tensor;

Element-wise inverse hyperbolic tangent.

cbrt

export declare function cbrt(t: Tensor): Tensor;

Element-wise cube root.

ceil

export declare function ceil(t: Tensor): Tensor;

Element-wise ceil (round up).

clip

export declare function clip(t: Tensor, min?: number, max?: number): Tensor;

Clip (limit) values in tensor.

cos

export declare function cos(t: Tensor): Tensor;

Element-wise cosine.

cosh

export declare function cosh(t: Tensor): Tensor;

Element-wise hyperbolic cosine.

cumprod

export declare function cumprod(t: Tensor, axis?: Axis): Tensor;

Cumulative product along axis.

cumsum

export declare function cumsum(t: Tensor, axis?: Axis): Tensor;

Cumulative sum along axis.

diff

export declare function diff(t: Tensor, n?: number, axis?: number): Tensor;

Calculate differences along axis.

div

export declare function div(a: Tensor, b: Tensor): Tensor;

Element-wise division.

equal

export declare function equal(a: Tensor, b: Tensor): Tensor;

Element-wise equality.

exp

export declare function exp(t: Tensor): Tensor;

Element-wise exponential.

exp2

export declare function exp2(t: Tensor): Tensor;

Element-wise base-2 exponential.

expm1

export declare function expm1(t: Tensor): Tensor;

Element-wise exp(x) - 1.

floor

export declare function floor(t: Tensor): Tensor;

Element-wise floor (round down).

floorDiv

export declare function floorDiv(a: Tensor, b: Tensor): Tensor;

Element-wise floor division.

greater

export declare function greater(a: Tensor, b: Tensor): Tensor;

Element-wise greater than comparison (a > b).

greaterEqual

export declare function greaterEqual(a: Tensor, b: Tensor): Tensor;

Element-wise greater than or equal comparison (a >= b).

isclose

export declare function isclose(a: Tensor, b: Tensor, rtol?: number, atol?: number): Tensor;

isclose is exported by deepbox/ndarray.

isfinite

export declare function isfinite(t: Tensor): Tensor;

Element-wise test for finite values (not NaN, not Inf).

isinf

export declare function isinf(t: Tensor): Tensor;

Element-wise test for infinity (+Inf or -Inf).

isnan

export declare function isnan(t: Tensor): Tensor;

Element-wise test for NaN (Not a Number) values.

less

export declare function less(a: Tensor, b: Tensor): Tensor;

Element-wise less than comparison (a < b).

lessEqual

export declare function lessEqual(a: Tensor, b: Tensor): Tensor;

Element-wise less than or equal comparison (a <= b).

log

export declare function log(t: Tensor): Tensor;

Element-wise natural logarithm.

log1p

export declare function log1p(t: Tensor): Tensor;

Element-wise log(1 + x).

log2

export declare function log2(t: Tensor): Tensor;

Element-wise base-2 logarithm.

log10

export declare function log10(t: Tensor): Tensor;

Element-wise base-10 logarithm.

logicalAnd

export declare function logicalAnd(a: Tensor, b: Tensor): Tensor;

logicalAnd is exported by deepbox/ndarray.

logicalNot

export declare function logicalNot(t: Tensor): Tensor;

Element-wise logical NOT.

logicalOr

export declare function logicalOr(a: Tensor, b: Tensor): Tensor;

Element-wise logical OR.

logicalXor

export declare function logicalXor(a: Tensor, b: Tensor): Tensor;

Element-wise logical XOR (exclusive OR).

max

export declare function max(t: Tensor, axis?: Axis | Axis[], keepdims?: boolean): Tensor;

Maximum value along axis.

maximum

export declare function maximum(a: Tensor, b: Tensor): Tensor;

Element-wise maximum of two tensors.

mean

export declare function mean(t: Tensor, axis?: Axis, keepdims?: boolean): Tensor;

Compute the arithmetic mean along the specified axis.

median

export declare function median(t: Tensor, axis?: Axis, keepdims?: boolean): Tensor;

median is exported by deepbox/ndarray.

min

export declare function min(t: Tensor, axis?: Axis | Axis[], keepdims?: boolean): Tensor;

Minimum value along axis.

minimum

export declare function minimum(a: Tensor, b: Tensor): Tensor;

Element-wise minimum of two tensors.

mod

export declare function mod(a: Tensor, b: Tensor): Tensor;

Element-wise modulo operation.

mul

export declare function mul(a: Tensor, b: Tensor): Tensor;

Element-wise multiplication.

mulScalar

export declare function mulScalar(t: Tensor, s: number): Tensor;

Multiply all elements of a tensor by a scalar value.

neg

export declare function neg(t: Tensor): Tensor;

Element-wise negation.

notEqual

export declare function notEqual(a: Tensor, b: Tensor): Tensor;

Element-wise inequality (not equal) comparison.

pow

export declare function pow(a: Tensor, b: Tensor): Tensor;

Element-wise power.

prod

export declare function prod(t: Tensor, axis?: Axis | Axis[], keepdims?: boolean): Tensor;

Product of array elements along axis.

reciprocal

export declare function reciprocal(t: Tensor): Tensor;

Element-wise reciprocal.

round

export declare function round(t: Tensor): Tensor;

Element-wise round to nearest integer.

rsqrt

export declare function rsqrt(t: Tensor): Tensor;

Element-wise reciprocal square root.

sign

export declare function sign(t: Tensor): Tensor;

Element-wise sign function.

sin

export declare function sin(t: Tensor): Tensor;

Element-wise sine.

sinh

export declare function sinh(t: Tensor): Tensor;

Element-wise hyperbolic sine.

sort

export declare function sort(t: Tensor, axis?: Axis | undefined, descending?: boolean): Tensor;

Sort values along a given axis.

sqrt

export declare function sqrt(t: Tensor): Tensor;

Element-wise square root.

square

export declare function square(t: Tensor): Tensor;

Element-wise square.

std

export declare function std(t: Tensor, axis?: Axis, keepdims?: boolean, ddof?: number): Tensor;

Standard deviation along axis.

sub

export declare function sub(a: Tensor, b: Tensor): Tensor;

Element-wise subtraction.

sum

export declare function sum(t: Tensor, axis?: Axis, keepdims?: boolean): Tensor;

Sum reduction.

tan

export declare function tan(t: Tensor): Tensor;

Element-wise tangent.

tanh

export declare function tanh(t: Tensor): Tensor;

Element-wise hyperbolic tangent.

trunc

export declare function trunc(t: Tensor): Tensor;

Element-wise truncate (round toward zero).

variance

export declare function variance(t: Tensor, axis?: Axis, keepdims?: boolean, ddof?: number): Tensor;

Variance along axis.

ndarray-ops.ts

import {  add,  exp,  greater,  logicalAnd,  mean,  mul,  sin,  sort,  std,  sum,  tensor,  where,} from "deepbox/ndarray";const x = tensor([[1, 2], [3, 4]]);const y = tensor([[10, 20], [30, 40]]);console.log(add(x, y).toString());console.log(mul(x, y).toString());console.log(exp(x).toString());console.log(sin(x).toString());console.log(sort(tensor([3, 1, 2])).toString());console.log(logicalAnd(tensor([1, 0, 1]), tensor([1, 1, 0])).toString());console.log(sum(x).toString(), mean(y).toString(), std(y).toString());console.log(where(greater(x, tensor(2)), x, y).toString());

Tensor Creation

Activation Functions