import type { Type as $Type, ICopy, IEqualsDelta, IEquiv, ILength, IRelease, Maybe, NumericArray } from "@thi.ng/api"; export interface TensorData extends Iterable, ILength { [id: number]: T; fill(x: T, start?: number, end?: number): TensorData; } export type Type = $Type | "num" | "str"; export type NumType = $Type | "num"; export type Shape0 = []; export type Shape1 = [number]; export type Shape2 = [number, number]; export type Shape3 = [number, number, number]; export type Shape4 = [number, number, number, number]; export type Shape = Shape0 | Shape1 | Shape2 | Shape3 | Shape4; export type ShapeTensor = S extends Shape4 ? ITensor4 : S extends Shape3 ? ITensor3 : S extends Shape2 ? ITensor2 : S extends Shape1 ? ITensor1 : ITensor0; export type Nested = T[] | T[][] | T[][][] | T[][][][]; export type NestedTensor, T> = N extends T[][][][] ? ITensor4 : N extends T[][][] ? ITensor3 : N extends T[][] ? ITensor2 : ITensor1; export interface TypeMap { u8: number; u8c: number; i8: number; u16: number; i16: number; u32: number; i32: number; f32: number; f64: number; num: number; str: string; } export interface TensorOpts { /** * Tensor data. Unless {@link TensorOpts.copy} is false, by default will be * copied to memory obtained from configured storage. */ data?: TensorData; /** * Optionally configured storage provider. By default uses the * datatype-specific implementation from global {@link STORAGE} registry. */ storage?: ITensorStorage; /** * Optionally configured stride tuple. By default the strides will be * obtained from the tensor shape and will be in row-major order. */ stride?: S; /** * Optional start index of the data values (only inteded to be used if * {@link TensorOpts.data} is given). */ offset?: number; /** * Only used if {@link TensorOpts.data} is given. If true (default), the * data will be copied to memory obtained from configured storage. */ copy?: boolean; } export interface TensorFromArrayOpts { type: T; storage?: ITensorStorage; } /** * Source data type for tensor conversion via {@link asTensor}. */ export interface TensorLike { /** Data type */ type: T; /** Tensor data/values (MUST match `type`) */ data: TensorData; /** Tensor shape */ shape: S; /** Stride/layout information of data */ stride: S; /** Start index (default: 0) */ offset?: number; } export interface ITensor extends ICopy>, IEquiv, IEqualsDelta>, IRelease { readonly type: Type; readonly storage: ITensorStorage; readonly data: TensorData; readonly shape: number[]; readonly stride: number[]; readonly offset: number; readonly length: number; readonly dim: number; readonly order: number[]; orderedShape: number[]; orderedStride: number[]; [Symbol.iterator](): IterableIterator; /** * Internal use only. Creates a shallow view used for broadcasting * operators. See {@link broadcast} for details. * * @param shape * @param stride * * @internal */ broadcast(shape: S, stride: S): ShapeTensor; /** * Returns a new tensor of same shape, but all values zeroed. Unless * `storage` is given, the new data will be allocated using this tensor's * storage provider. * * @param storage */ empty(storage?: ITensorStorage): this; fill(x: T): this; /** * Computes linear array index from given grid position. Reverse-op of * {@link ITensor.position}. * * @remarks * The given `pos` is assumed to be integral and valid. No bounds checking * performed. * * @param pos */ index(pos: NumericArray): number; /** * Returns an iterator of array indices used by this tensor */ indices(): IterableIterator; /** * Computes nD grid position for given linear array index. Reverse-op of * {@link ITensor.index}. * * @remarks * The given `index` is assumed to be valid. No bounds checking performed. * * **CAUTION:** Currently only supports tensors with positive strides, * otherwise will yield incorrect results! Tensors with negative strides * (aka flipped axes in reverse order) need to be first packed via * {@link ITensor.pack}. * * @param index */ position(index: number): number[]; /** * Returns value at given grid position. No bounds checking. * * @param pos */ get(pos: NumericArray): T; /** * Sets value at given grid position. No bounds checking. * * @param pos * @param value */ set(pos: NumericArray, value: T): this; /** * Returns a new tensor of the bottom-right region starting from given * `pos`. View transform only, no data will be copied. * * @remarks * Also see {@link Itensor.hi}, {@link ITensor.crop}. * * @example * ```ts tangle:../export/itensor-lo.ts * import { print, range } from "@thi.ng/tensors"; * * const a = range(16).reshape([4, 4]); * print(a); * // 0 1.0000 2.0000 3.0000 * // 4.0000 5.0000 6.0000 7.0000 * // 8.0000 9.0000 10.0000 11.0000 * // 12.0000 13.0000 14.0000 15.0000 * * const b = a.lo([2, 1]); * print(b); * // 9.0000 10.0000 11.0000 * // 13.0000 14.0000 15.0000 * ``` * * @param pos */ lo(pos: NumericArray): this; /** * Returns a new tensor of the top-left region until given `pos` (excluded). * View transform only, no data will be copied. * * @remarks * Also see {@link Itensor.lo}, {@link ITensor.crop}. * * @example * ```ts tangle:../export/itensor-hi.ts * import { print, range } from "@thi.ng/tensors"; * * const a = range(16).reshape([4, 4]); * print(a); * // 0 1.0000 2.0000 3.0000 * // 4.0000 5.0000 6.0000 7.0000 * // 8.0000 9.0000 10.0000 11.0000 * // 12.0000 13.0000 14.0000 15.0000 * * const b = a.hi([2, 3]); * print(b); * // 0 1.0000 2.0000 * // 4.0000 5.0000 6.0000 * ``` * * @param pos */ hi(pos: NumericArray): this; /** * Returns a new tensor of the extracted region defined by `pos` and `size`. * This op is a combination of {@link ITensor.lo} and {@link ITensor.hi}. * View transform only, no data will be copied. * * @example * ```ts tangle:../export/itensor-crop.ts * import { print, range } from "@thi.ng/tensors"; * * const a = range(16).reshape([4, 4]); * print(a); * // 0 1.0000 2.0000 3.0000 * // 4.0000 5.0000 6.0000 7.0000 * // 8.0000 9.0000 10.0000 11.0000 * // 12.0000 13.0000 14.0000 15.0000 * * const b = a.crop([1, 1], [2, 2]); * print(b); * // 5.0000 6.0000 * // 9.0000 10.0000 * ``` * * @param pos * @param size */ crop(pos: NumericArray, size: NumericArray): this; /** * Returns a new tensor with step sizes adjusted for selected axes (Using * zero for an axis will keep its current step size). View transform only, * no data will be copied. * * @example * ```ts tangle:../export/itensor-step.ts * import { print, range } from "@thi.ng/tensors"; * * const a = range(16).reshape([4, 4]); * print(a); * // 0 1.0000 2.0000 3.0000 * // 4.0000 5.0000 6.0000 7.0000 * // 8.0000 9.0000 10.0000 11.0000 * // 12.0000 13.0000 14.0000 15.0000 * * // only select every 2nd row * const b = a.step([2, 1]); * print(b); * // 0 1.0000 2.0000 3.0000 * // 8.0000 9.0000 10.0000 11.0000 * * // keep rows as is (zero), only select every 2nd column * print(b.step([0, 2])); * // 0 2.0000 * // 8.0000 10.0000 * ``` * * @param select */ step(select: NumericArray): this; /** * Returns a new tensor with only the `select`ed axes. A -1 will select all * value in that axis. View transform only, no data will be copied. * * @example * ```ts tangle:../export/itensor-pick.ts * import { print, range } from "@thi.ng/tensors"; * * // 3D 4x4x4 tensor with values in [0,64) range * const a = range(64).reshape([4, 4, 4]); * * // pick entire slice #2 * print(a.pick([2])); * // 32.0000 33.0000 34.0000 35.0000 * // 36.0000 37.0000 38.0000 39.0000 * // 40.0000 41.0000 42.0000 43.0000 * // 44.0000 45.0000 46.0000 47.0000 * * // pick slice #2, row #2 (1D tensor) * print(a.pick([2, 2])); * // 40.0000 41.0000 42.0000 43.0000 * * // pick slice #2, column #2 (1D tensor) * print(a.pick([2, -1, 2])); * // 34.0000 38.0000 42.0000 46.0000 * ``` * * @param select */ pick(select: NumericArray): ITensor; /** * Creates a "packed" copy of this tensor with dense striding and the new * data array only holding the values actually referenced by this tensor. * Unless `storage` is given, the new data will be allocated using this * tensor's storage provider. * * @remarks * Since most other `ITensor` ops are zero-copy, view-only transforms, often * resulting in "sparse" views which are only addressing a subset of the * values stored, using `.pack()` is useful to extract data into a dense * tensor/buffer. * * @example * ```ts tangle:../export/itensor-pack.ts * import { range } from "@thi.ng/tensors"; * * const a = range(16).reshape([4, 4]); * console.log("a data", a.data); * // a data [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 ] * * // only select every 2nd row & column * const b = a.step([2, 2]); * console.log("b values", [...b]); * // b values [ 0, 2, 8, 10 ] * console.log("b data", b.data); * // b data [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 ] * * // create packed version of `b` * const c = b.pack(); * console.log("c data", c.data); * // c data [ 0, 2, 8, 10 ] * ``` * * @param storage */ pack(storage?: ITensorStorage): this; /** * Returns a new tensor with same data but given new shape (and optionally * new strides). The total number of elements of the new shape MUST match * that of the current shape (otherwise an error will be thrown). * * @remarks * **IMPORTANT:** If no `newStride` is given, only "dense" tensors can be * reshaped reliably (if in doubt, use {@link ITensor.pack} prior to * reshaping). * * @remarks * Also see {@link ITensor.crop} and {@link ITensor.resize} * * @example * ```ts tangle:../export/itensor-reshape.ts * import { print, range } from "@thi.ng/tensors"; * * // 1D tensor * const a = range(16); * * // reshape as 2D tensor * print(a.reshape([4, 4])); * // 0 1.0000 2.0000 3.0000 * // 4.0000 5.0000 6.0000 7.0000 * // 8.0000 9.0000 10.0000 11.0000 * // 12.0000 13.0000 14.0000 15.0000 * * // reshape as 3D tensor * print(a.reshape([2, 2, 4])); * // --- 0: --- * // 0 1.0000 2.0000 3.0000 * // 4.0000 5.0000 6.0000 7.0000 * // --- 1: --- * // 8.0000 9.0000 10.0000 11.0000 * // 12.0000 13.0000 14.0000 15.0000 * ``` * * @param newShape * @param newStride */ reshape(newShape: S, newStride?: S): ShapeTensor; /** * Returns a copy of the tensor resized to `newShape`. If the new shape is * larger than the current shape, the extra data values will be initialized * to `fill` (default: zero). Values will be copied in current iteration * order (same logic as numpy). Unless `storage` is given, the new data will * be allocated using this tensor's storage provider. * * @remarks * Also see {@link ITensor.crop}, {@link ITensor.reshape}. * * @example * ```ts tangle:../export/itensor-resize.ts * import { print, range } from "@thi.ng/tensors"; * * // 2D 4x4 tensor with values in [0,16) range * const a = range(16).reshape([4, 4]); * * print(a.resize([4, 8])); * // 0 1.0000 2.0000 3.0000 4.0000 5.0000 6.0000 7.0000 * // 8.0000 9.0000 10.0000 11.0000 12.0000 13.0000 14.0000 15.0000 * // 0 0 0 0 0 0 0 0 * // 0 0 0 0 0 0 0 0 * ``` * * @param newShape * @param fill * @param storage */ resize(newShape: S, fill?: T, storage?: ITensorStorage): ShapeTensor; /** * Returns a new tensor with the given new axis `order`. View transform * only, no data will be copied. * * @example * ```ts tangle:../export/itensor-transpose.ts * import { print, range } from "@thi.ng/tensors"; * * const a = range(12).reshape([3, 4]); * print(a); * // 0 1.0000 2.0000 3.0000 * // 4.0000 5.0000 6.0000 7.0000 * // 8.0000 9.0000 10.0000 11.0000 * * // swap row & column order * const b = a.transpose([1, 0]); * print(b); * // 0 4.0000 8.0000 * // 1.0000 5.0000 9.0000 * // 2.0000 6.0000 10.0000 * // 3.0000 7.0000 11.0000 * ``` * * @param order */ transpose(order: NumericArray): this; toJSON(): any; } export type ITensor0 = ITensor & { dim: 0; }; export type ITensor1 = ITensor & { dim: 1; }; export type ITensor2 = ITensor & { dim: 2; }; export type ITensor3 = ITensor & { dim: 3; }; export type ITensor4 = ITensor & { dim: 4; }; export interface TensorCtor { new (type: Type, storage: ITensorStorage, data: TensorData, shape: number[], stride: number[], offset?: number): ITensor; } export interface ITensorStorage { /** * Attempts to allocate/create an array for given number of items. Throws an * error if unsuccessful. * * @param size */ alloc(size: number): TensorData; /** * Attempts to allocate/create an array for given iterable. Throws an * error if unsuccessful. * * @param iter */ from(iter: Iterable): TensorData; /** * Attempts to release the array/memory used by given buffer. Returns true * if successful. * * @param buf */ release(buf: TensorData): boolean; } export type StorageRegistry = Record>; export interface TensorOpT { (out: ITensor0 | null, a: ITensor0): ITensor0; (out: ITensor1 | null, a: ITensor1): ITensor1; (out: ITensor2 | null, a: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor4): ITensor4; } export interface TensorOpTT { (out: ITensor0 | null, a: ITensor0, b: ITensor0): ITensor0; (out: ITensor1 | null, a: ITensor0, b: ITensor1): ITensor1; (out: ITensor2 | null, a: ITensor0, b: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor0, b: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor0, b: ITensor4): ITensor4; (out: ITensor1 | null, a: ITensor1, b: ITensor0): ITensor1; (out: ITensor1 | null, a: ITensor1, b: ITensor1): ITensor1; (out: ITensor2 | null, a: ITensor1, b: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor1, b: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor1, b: ITensor4): ITensor4; (out: ITensor2 | null, a: ITensor2, b: ITensor0): ITensor2; (out: ITensor2 | null, a: ITensor2, b: ITensor1): ITensor2; (out: ITensor2 | null, a: ITensor2, b: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor2, b: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor2, b: ITensor4): ITensor4; (out: ITensor3 | null, a: ITensor3, b: ITensor0): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor1): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor2): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor3, b: ITensor4): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor0): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor1): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor2): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor3): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor4): ITensor4; } export interface TensorOpTTT { (out: ITensor0 | null, a: ITensor0, b: ITensor0, c: ITensor0): ITensor0; (out: ITensor1 | null, a: ITensor0, b: ITensor0, c: ITensor1): ITensor1; (out: ITensor2 | null, a: ITensor0, b: ITensor0, c: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor0, b: ITensor0, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor0, b: ITensor0, c: ITensor4): ITensor4; (out: ITensor1 | null, a: ITensor0, b: ITensor1, c: ITensor0): ITensor1; (out: ITensor1 | null, a: ITensor0, b: ITensor1, c: ITensor1): ITensor1; (out: ITensor2 | null, a: ITensor0, b: ITensor1, c: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor0, b: ITensor1, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor0, b: ITensor1, c: ITensor4): ITensor4; (out: ITensor2 | null, a: ITensor0, b: ITensor2, c: ITensor0): ITensor2; (out: ITensor2 | null, a: ITensor0, b: ITensor2, c: ITensor1): ITensor2; (out: ITensor2 | null, a: ITensor0, b: ITensor2, c: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor0, b: ITensor2, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor0, b: ITensor2, c: ITensor4): ITensor4; (out: ITensor3 | null, a: ITensor0, b: ITensor3, c: ITensor0): ITensor3; (out: ITensor3 | null, a: ITensor0, b: ITensor3, c: ITensor1): ITensor3; (out: ITensor3 | null, a: ITensor0, b: ITensor3, c: ITensor2): ITensor3; (out: ITensor3 | null, a: ITensor0, b: ITensor3, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor0, b: ITensor3, c: ITensor4): ITensor4; (out: ITensor4 | null, a: ITensor0, b: ITensor4, c: ITensor0): ITensor4; (out: ITensor4 | null, a: ITensor0, b: ITensor4, c: ITensor1): ITensor4; (out: ITensor4 | null, a: ITensor0, b: ITensor4, c: ITensor2): ITensor4; (out: ITensor4 | null, a: ITensor0, b: ITensor4, c: ITensor3): ITensor4; (out: ITensor4 | null, a: ITensor0, b: ITensor4, c: ITensor4): ITensor4; (out: ITensor1 | null, a: ITensor1, b: ITensor0, c: ITensor0): ITensor1; (out: ITensor1 | null, a: ITensor1, b: ITensor0, c: ITensor1): ITensor1; (out: ITensor2 | null, a: ITensor1, b: ITensor0, c: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor1, b: ITensor0, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor1, b: ITensor0, c: ITensor4): ITensor4; (out: ITensor1 | null, a: ITensor1, b: ITensor1, c: ITensor0): ITensor1; (out: ITensor1 | null, a: ITensor1, b: ITensor1, c: ITensor1): ITensor1; (out: ITensor2 | null, a: ITensor1, b: ITensor1, c: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor1, b: ITensor1, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor1, b: ITensor1, c: ITensor4): ITensor4; (out: ITensor2 | null, a: ITensor1, b: ITensor2, c: ITensor0): ITensor2; (out: ITensor2 | null, a: ITensor1, b: ITensor2, c: ITensor1): ITensor2; (out: ITensor2 | null, a: ITensor1, b: ITensor2, c: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor1, b: ITensor2, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor1, b: ITensor2, c: ITensor4): ITensor4; (out: ITensor3 | null, a: ITensor1, b: ITensor3, c: ITensor0): ITensor3; (out: ITensor3 | null, a: ITensor1, b: ITensor3, c: ITensor1): ITensor3; (out: ITensor3 | null, a: ITensor1, b: ITensor3, c: ITensor2): ITensor3; (out: ITensor3 | null, a: ITensor1, b: ITensor3, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor1, b: ITensor3, c: ITensor4): ITensor4; (out: ITensor4 | null, a: ITensor1, b: ITensor4, c: ITensor0): ITensor4; (out: ITensor4 | null, a: ITensor1, b: ITensor4, c: ITensor1): ITensor4; (out: ITensor4 | null, a: ITensor1, b: ITensor4, c: ITensor2): ITensor4; (out: ITensor4 | null, a: ITensor1, b: ITensor4, c: ITensor3): ITensor4; (out: ITensor4 | null, a: ITensor1, b: ITensor4, c: ITensor4): ITensor4; (out: ITensor2 | null, a: ITensor2, b: ITensor0, c: ITensor0): ITensor2; (out: ITensor2 | null, a: ITensor2, b: ITensor0, c: ITensor1): ITensor2; (out: ITensor2 | null, a: ITensor2, b: ITensor0, c: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor2, b: ITensor0, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor2, b: ITensor0, c: ITensor4): ITensor4; (out: ITensor2 | null, a: ITensor2, b: ITensor1, c: ITensor0): ITensor2; (out: ITensor2 | null, a: ITensor2, b: ITensor1, c: ITensor1): ITensor2; (out: ITensor2 | null, a: ITensor2, b: ITensor1, c: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor2, b: ITensor1, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor2, b: ITensor1, c: ITensor4): ITensor4; (out: ITensor2 | null, a: ITensor2, b: ITensor2, c: ITensor0): ITensor2; (out: ITensor2 | null, a: ITensor2, b: ITensor2, c: ITensor1): ITensor2; (out: ITensor2 | null, a: ITensor2, b: ITensor2, c: ITensor2): ITensor2; (out: ITensor3 | null, a: ITensor2, b: ITensor2, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor2, b: ITensor2, c: ITensor4): ITensor4; (out: ITensor3 | null, a: ITensor2, b: ITensor3, c: ITensor0): ITensor3; (out: ITensor3 | null, a: ITensor2, b: ITensor3, c: ITensor1): ITensor3; (out: ITensor3 | null, a: ITensor2, b: ITensor3, c: ITensor2): ITensor3; (out: ITensor3 | null, a: ITensor2, b: ITensor3, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor2, b: ITensor3, c: ITensor4): ITensor4; (out: ITensor4 | null, a: ITensor2, b: ITensor4, c: ITensor0): ITensor4; (out: ITensor4 | null, a: ITensor2, b: ITensor4, c: ITensor1): ITensor4; (out: ITensor4 | null, a: ITensor2, b: ITensor4, c: ITensor2): ITensor4; (out: ITensor4 | null, a: ITensor2, b: ITensor4, c: ITensor3): ITensor4; (out: ITensor4 | null, a: ITensor2, b: ITensor4, c: ITensor4): ITensor4; (out: ITensor3 | null, a: ITensor3, b: ITensor0, c: ITensor0): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor0, c: ITensor1): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor0, c: ITensor2): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor0, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor3, b: ITensor0, c: ITensor4): ITensor4; (out: ITensor3 | null, a: ITensor3, b: ITensor1, c: ITensor0): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor1, c: ITensor1): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor1, c: ITensor2): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor1, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor3, b: ITensor1, c: ITensor4): ITensor4; (out: ITensor3 | null, a: ITensor3, b: ITensor2, c: ITensor0): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor2, c: ITensor1): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor2, c: ITensor2): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor2, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor3, b: ITensor2, c: ITensor4): ITensor4; (out: ITensor3 | null, a: ITensor3, b: ITensor3, c: ITensor0): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor3, c: ITensor1): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor3, c: ITensor2): ITensor3; (out: ITensor3 | null, a: ITensor3, b: ITensor3, c: ITensor3): ITensor3; (out: ITensor4 | null, a: ITensor3, b: ITensor3, c: ITensor4): ITensor4; (out: ITensor4 | null, a: ITensor3, b: ITensor4, c: ITensor0): ITensor4; (out: ITensor4 | null, a: ITensor3, b: ITensor4, c: ITensor1): ITensor4; (out: ITensor4 | null, a: ITensor3, b: ITensor4, c: ITensor2): ITensor4; (out: ITensor4 | null, a: ITensor3, b: ITensor4, c: ITensor3): ITensor4; (out: ITensor4 | null, a: ITensor3, b: ITensor4, c: ITensor4): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor0, c: ITensor0): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor0, c: ITensor1): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor0, c: ITensor2): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor0, c: ITensor3): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor0, c: ITensor4): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor1, c: ITensor0): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor1, c: ITensor1): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor1, c: ITensor2): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor1, c: ITensor3): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor1, c: ITensor4): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor2, c: ITensor0): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor2, c: ITensor1): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor2, c: ITensor2): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor2, c: ITensor3): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor2, c: ITensor4): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor3, c: ITensor0): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor3, c: ITensor1): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor3, c: ITensor2): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor3, c: ITensor3): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor3, c: ITensor4): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor4, c: ITensor0): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor4, c: ITensor1): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor4, c: ITensor2): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor4, c: ITensor3): ITensor4; (out: ITensor4 | null, a: ITensor4, b: ITensor4, c: ITensor4): ITensor4; } export interface TensorOpN { (out: ITensor0, n: A): ITensor0; (out: ITensor1, n: A): ITensor1; (out: ITensor2, n: A): ITensor2; (out: ITensor3, n: A): ITensor3; (out: ITensor4, n: A): ITensor4; } export interface TensorOpTN { (out: ITensor0 | null, a: ITensor0, n: T): ITensor0; (out: ITensor1 | null, a: ITensor1, n: T): ITensor1; (out: ITensor2 | null, a: ITensor2, n: T): ITensor2; (out: ITensor3 | null, a: ITensor3, n: T): ITensor3; (out: ITensor4 | null, a: ITensor4, n: T): ITensor4; } export interface TensorOpTNO { (out: ITensor0 | null, a: ITensor0, n?: T): ITensor0; (out: ITensor1 | null, a: ITensor1, n?: T): ITensor1; (out: ITensor2 | null, a: ITensor2, n?: T): ITensor2; (out: ITensor3 | null, a: ITensor3, n?: T): ITensor3; (out: ITensor4 | null, a: ITensor4, n?: T): ITensor4; } export interface TensorOpTNN { (out: ITensor0, a: ITensor0, n: T, m: T): ITensor0; (out: ITensor1, a: ITensor1, n: T, m: T): ITensor1; (out: ITensor2, a: ITensor2, n: T, m: T): ITensor2; (out: ITensor3, a: ITensor3, n: T, m: T): ITensor3; (out: ITensor4, a: ITensor4, n: T, m: T): ITensor4; } export type TensorOpRT = ITensor> = (a: TA) => B; export type TensorOpRTT = ITensor> = (a: TA, b: TA) => B; export interface MultiTensorOp { /** * Adds / overwrites implementation for given tensor dimension. * * @param dim - * @param op - */ add(dim: number, op: TOP): TOP; /** * Adds / overwrites default implementation (SHOULD support arbitrary tensor * dimensions). * * @param op - */ default(op: TOP): TOP; /** * Returns implementation for given tensor dimension or default * implementation. * * @param dim - */ impl(dim?: number): Maybe; } export type MultiTensorOpImpl = T & MultiTensorOp; /** * Convolution kernel spec for use with {@link applyKernel}. * * @remarks * Provided implementations: * * - {@link MAX2_MOORE} * - {@link MAX2_VON_NEUMANN} * - {@link MAXIMA2_MOORE} * - {@link MAXIMA2_VON_NEUMANN} */ export interface KernelSpec { /** * Kernel shape/size */ shape: Exclude; /** * Windowed intialization. Returns initial accumulator for each new kernel * window. */ init: () => T; /** * Windowed reduction function. Receives current accumulator, domain value * and kernel-local coordinates. Returns updated accumulator. * * @param acc * @param value * @param coords */ reduce: (acc: T, value: number, ...coords: number[]) => T; /** * Windowed reducer result function. Produces final result from current * accumulator. * * @param acc */ complete: (acc: T) => number; } //# sourceMappingURL=api.d.ts.map