@thi.ng/tensors 0.1.0

package/defopttt.js

@@ -0,0 +1,177 @@
+import { top } from "./top.js";
+const defOpTTT = (fn, dispatch = 1) => {
+  const f1 = (out, a, b, c) => {
+    !out && (out = a);
+    const {
+      data: odata,
+      offset: oo,
+      stride: [txo]
+    } = out;
+    const {
+      data: adata,
+      offset: oa,
+      shape: [sx],
+      stride: [txa]
+    } = a;
+    const {
+      data: bdata,
+      offset: ob,
+      stride: [txb]
+    } = b;
+    const {
+      data: cdata,
+      offset: oc,
+      stride: [txc]
+    } = c;
+    for (let x = 0; x < sx; x++) {
+      odata[oo + x * txo] = fn(
+        adata[oa + x * txa],
+        bdata[ob + x * txb],
+        cdata[oc + x * txc]
+      );
+    }
+    return out;
+  };
+  const f2 = (out, a, b, c) => {
+    !out && (out = a);
+    const {
+      data: odata,
+      offset: oo,
+      stride: [txo, tyo]
+    } = out;
+    const {
+      data: adata,
+      offset: oa,
+      shape: [sx, sy],
+      stride: [txa, tya]
+    } = a;
+    const {
+      data: bdata,
+      offset: ob,
+      stride: [txb, tyb]
+    } = b;
+    const {
+      data: cdata,
+      offset: oc,
+      stride: [txc, tyc]
+    } = c;
+    for (let x = 0; x < sx; x++) {
+      const oox = oo + x * txo;
+      const oax = oa + x * txa;
+      const obx = ob + x * txb;
+      const ocx = oc + x * txc;
+      for (let y = 0; y < sy; y++) {
+        odata[oox + y * tyo] = fn(
+          adata[oax + y * tya],
+          bdata[obx + y * tyb],
+          cdata[ocx + y * tyc]
+        );
+      }
+    }
+    return out;
+  };
+  const f3 = (out, a, b, c) => {
+    !out && (out = a);
+    const {
+      data: odata,
+      offset: oo,
+      stride: [txo, tyo, tzo]
+    } = out;
+    const {
+      data: adata,
+      offset: oa,
+      shape: [sx, sy, sz],
+      stride: [txa, tya, tza]
+    } = a;
+    const {
+      data: bdata,
+      offset: ob,
+      stride: [txb, tyb, tzb]
+    } = b;
+    const {
+      data: cdata,
+      offset: oc,
+      stride: [txc, tyc, tzc]
+    } = c;
+    for (let x = 0; x < sx; x++) {
+      const oox = oo + x * txo;
+      const oax = oa + x * txa;
+      const obx = ob + x * txb;
+      const ocx = oc + x * txc;
+      for (let y = 0; y < sy; y++) {
+        const ooy = oox + y * tyo;
+        const oay = oax + y * tya;
+        const oby = obx + y * tyb;
+        const ocy = ocx + y * tyc;
+        for (let z = 0; z < sz; z++) {
+          odata[ooy + z * tzo] = fn(
+            adata[oay + z * tza],
+            bdata[oby + z * tzb],
+            cdata[ocy + z * tzc]
+          );
+        }
+      }
+    }
+    return out;
+  };
+  const f4 = (out, a, b, c) => {
+    !out && (out = a);
+    const {
+      data: odata,
+      offset: oo,
+      stride: [txo, tyo, tzo, two]
+    } = out;
+    const {
+      data: adata,
+      offset: oa,
+      shape: [sx, sy, sz, sw],
+      stride: [txa, tya, tza, twa]
+    } = a;
+    const {
+      data: bdata,
+      offset: ob,
+      stride: [txb, tyb, tzb, twb]
+    } = b;
+    const {
+      data: cdata,
+      offset: oc,
+      stride: [txc, tyc, tzc, twc]
+    } = c;
+    for (let x = 0; x < sx; x++) {
+      const oox = oo + x * txo;
+      const oax = oa + x * txa;
+      const obx = ob + x * txb;
+      const ocx = oc + x * txc;
+      for (let y = 0; y < sy; y++) {
+        const ooy = oox + y * tyo;
+        const oay = oax + y * tya;
+        const oby = obx + y * tyb;
+        const ocy = ocx + y * tyc;
+        for (let z = 0; z < sz; z++) {
+          const ooz = ooy + z * tzo;
+          const oaz = oay + z * tza;
+          const obz = oby + z * tzb;
+          const ocz = ocy + z * tzc;
+          for (let w = 0; w < sw; w++) {
+            odata[ooz + w * two] = fn(
+              adata[oaz + w * twa],
+              bdata[obz + w * twb],
+              cdata[ocz + w * twc]
+            );
+          }
+        }
+      }
+    }
+    return out;
+  };
+  return [
+    top(dispatch, void 0, f1, f2, f3, f4),
+    f1,
+    f2,
+    f3,
+    f4
+  ];
+};
+export {
+  defOpTTT
+};

package/div.d.ts

@@ -0,0 +1,42 @@
+/**
+ * Componentwise nD tensor division. Writes result to `out`. If `out` is null,
+ * mutates `a`. Multi-method.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ * @param n - scalar
+ */
+export declare const div: import("./api.js").MultiTensorOpTT<number>;
+/**
+ * Same as {@link div} for 1D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ * @param n - scalar
+ */
+export declare const div1: import("./api.js").TensorOpTT<number, number, import("./tensor.js").Tensor1<number>, import("./tensor.js").Tensor1<number>>;
+/**
+ * Same as {@link div} for 2D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ * @param n - scalar
+ */
+export declare const div2: import("./api.js").TensorOpTT<number, number, import("./tensor.js").Tensor2<number>, import("./tensor.js").Tensor2<number>>;
+/**
+ * Same as {@link div} for 3D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ * @param n - scalar
+ */
+export declare const div3: import("./api.js").TensorOpTT<number, number, import("./tensor.js").Tensor3<number>, import("./tensor.js").Tensor3<number>>;
+/**
+ * Same as {@link div} for 4D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ * @param n - scalar
+ */
+export declare const div4: import("./api.js").TensorOpTT<number, number, import("./tensor.js").Tensor4<number>, import("./tensor.js").Tensor4<number>>;
+//# sourceMappingURL=div.d.ts.map

package/div.js

@@ -0,0 +1,15 @@
+import { $div } from "@thi.ng/vectors/ops";
+import { defOpTT } from "./defoptt.js";
+const [a, b, c, d, e] = defOpTT($div);
+const div = a;
+const div1 = b;
+const div2 = c;
+const div3 = d;
+const div4 = e;
+export {
+  div,
+  div1,
+  div2,
+  div3,
+  div4
+};

package/divn.d.ts

@@ -0,0 +1,42 @@
+/**
+ * Componentwise nD tensor division with uniform scalar `n`. Writes result to
+ * `out`. If `out` is null, mutates `a`. Multi-method.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ * @param n - scalar
+ */
+export declare const divN: import("./api.js").MultiTensorOpTN<number>;
+/**
+ * Same as {@link divN} for 1D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ * @param n - scalar
+ */
+export declare const divN1: import("./api.js").TensorOpTN<number, number, import("./tensor.js").Tensor1<number>, import("./tensor.js").Tensor1<number>>;
+/**
+ * Same as {@link divN} for 2D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ * @param n - scalar
+ */
+export declare const divN2: import("./api.js").TensorOpTN<number, number, import("./tensor.js").Tensor2<number>, import("./tensor.js").Tensor2<number>>;
+/**
+ * Same as {@link divN} for 3D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ * @param n - scalar
+ */
+export declare const divN3: import("./api.js").TensorOpTN<number, number, import("./tensor.js").Tensor3<number>, import("./tensor.js").Tensor3<number>>;
+/**
+ * Same as {@link divN} for 4D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ * @param n - scalar
+ */
+export declare const divN4: import("./api.js").TensorOpTN<number, number, import("./tensor.js").Tensor4<number>, import("./tensor.js").Tensor4<number>>;
+//# sourceMappingURL=divn.d.ts.map

package/divn.js

@@ -0,0 +1,15 @@
+import { $div } from "@thi.ng/vectors/ops";
+import { defOpTN } from "./defoptn.js";
+const [a, b, c, d, e] = defOpTN($div);
+const divN = a;
+const divN1 = b;
+const divN2 = c;
+const divN3 = d;
+const divN4 = e;
+export {
+  divN,
+  divN1,
+  divN2,
+  divN3,
+  divN4
+};

package/dot.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Dot product of given nD tensor. Multi-method.
+ *
+ * @param a - input tensor
+ */
+export declare const dot: import("./api.js").MultiTensorOpRTT<number, number>;
+/**
+ * Dot product of given 1D tensor.
+ *
+ * @param a - input tensor
+ */
+export declare const dot1: import("./api.js").TensorOpRTT<number, number, import("./tensor.js").Tensor1<number>>;
+/**
+ * Dot product of given 2D tensor.
+ *
+ * @param a - input tensor
+ */
+export declare const dot2: import("./api.js").TensorOpRTT<number, number, import("./tensor.js").Tensor2<number>>;
+/**
+ * Dot product of given 3D tensor.
+ *
+ * @param a - input tensor
+ */
+export declare const dot3: import("./api.js").TensorOpRTT<number, number, import("./tensor.js").Tensor3<number>>;
+/**
+ * Dot product of given 3D tensor.
+ *
+ * @param a - input tensor
+ */
+export declare const dot4: import("./api.js").TensorOpRTT<number, number, import("./tensor.js").Tensor4<number>>;
+//# sourceMappingURL=dot.d.ts.map

package/dot.js

@@ -0,0 +1,17 @@
+import { defOpRTT } from "./defoprtt.js";
+const [a, b, c, d, e] = defOpRTT(
+  (acc, x, y) => acc + x * y,
+  () => 0
+);
+const dot = a;
+const dot1 = b;
+const dot2 = c;
+const dot3 = d;
+const dot4 = e;
+export {
+  dot,
+  dot1,
+  dot2,
+  dot3,
+  dot4
+};

package/errors.d.ts

@@ -0,0 +1,15 @@
+import type { NumericArray } from "@thi.ng/api";
+export declare const IllegalShapeError: {
+    new (msg?: NumericArray | undefined): {
+        origMessage: string;
+        name: string;
+        message: string;
+        stack?: string;
+        cause?: unknown;
+    };
+    captureStackTrace(targetObject: object, constructorOpt?: Function): void;
+    prepareStackTrace?: ((err: Error, stackTraces: NodeJS.CallSite[]) => any) | undefined;
+    stackTraceLimit: number;
+};
+export declare const illegalShape: (shape: NumericArray) => never;
+//# sourceMappingURL=errors.d.ts.map

package/errors.js

@@ -0,0 +1,9 @@
+import { defError } from "@thi.ng/errors/deferror";
+const IllegalShapeError = defError(() => "illegal shape");
+const illegalShape = (shape) => {
+  throw new IllegalShapeError(shape);
+};
+export {
+  IllegalShapeError,
+  illegalShape
+};

package/exp.d.ts

@@ -0,0 +1,37 @@
+/**
+ * Componentwise computes `Math.exp` of given nD tensor and writes result to
+ * `out`. If `out` is null, mutates original. Multi-method.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ */
+export declare const exp: import("./api.js").MultiTensorOpT<number>;
+/**
+ * Same as {@link exp} for 1D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ */
+export declare const exp_1: import("./api.js").TensorOpT<number, number, import("./tensor.js").Tensor1<number>, import("./tensor.js").Tensor1<number>>;
+/**
+ * Same as {@link exp} for 2D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ */
+export declare const exp_2: import("./api.js").TensorOpT<number, number, import("./tensor.js").Tensor2<number>, import("./tensor.js").Tensor2<number>>;
+/**
+ * Same as {@link exp} for 3D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ */
+export declare const exp_3: import("./api.js").TensorOpT<number, number, import("./tensor.js").Tensor3<number>, import("./tensor.js").Tensor3<number>>;
+/**
+ * Same as {@link exp} for 4D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ */
+export declare const exp_4: import("./api.js").TensorOpT<number, number, import("./tensor.js").Tensor4<number>, import("./tensor.js").Tensor4<number>>;
+//# sourceMappingURL=exp.d.ts.map

package/exp.js

@@ -0,0 +1,14 @@
+import { defOpT } from "./defopt.js";
+const [a, b, c, d, e] = defOpT(Math.exp);
+const exp = a;
+const exp_1 = b;
+const exp_2 = c;
+const exp_3 = d;
+const exp_4 = e;
+export {
+  exp,
+  exp_1,
+  exp_2,
+  exp_3,
+  exp_4
+};

package/exp2.d.ts

@@ -0,0 +1,37 @@
+/**
+ * Componentwise computes `2^x` of given nD tensor and writes result to
+ * `out`. If `out` is null, mutates original. Multi-method.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ */
+export declare const exp2: import("./api.js").MultiTensorOpT<number>;
+/**
+ * Same as {@link exp2} for 1D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ */
+export declare const exp2_1: import("./api.js").TensorOpT<number, number, import("./tensor.js").Tensor1<number>, import("./tensor.js").Tensor1<number>>;
+/**
+ * Same as {@link exp2} for 2D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ */
+export declare const exp2_2: import("./api.js").TensorOpT<number, number, import("./tensor.js").Tensor2<number>, import("./tensor.js").Tensor2<number>>;
+/**
+ * Same as {@link exp2} for 3D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ */
+export declare const exp2_3: import("./api.js").TensorOpT<number, number, import("./tensor.js").Tensor3<number>, import("./tensor.js").Tensor3<number>>;
+/**
+ * Same as {@link exp2} for 4D tensors.
+ *
+ * @param out - output tensor
+ * @param a - input tensor
+ */
+export declare const exp2_4: import("./api.js").TensorOpT<number, number, import("./tensor.js").Tensor4<number>, import("./tensor.js").Tensor4<number>>;
+//# sourceMappingURL=exp2.d.ts.map

package/exp2.js

@@ -0,0 +1,14 @@
+import { defOpT } from "./defopt.js";
+const [a, b, c, d, e] = defOpT((x) => 2 ** x);
+const exp2 = a;
+const exp2_1 = b;
+const exp2_2 = c;
+const exp2_3 = d;
+const exp2_4 = e;
+export {
+  exp2,
+  exp2_1,
+  exp2_2,
+  exp2_3,
+  exp2_4
+};

package/format.d.ts

@@ -0,0 +1,14 @@
+import type { ITensor } from "./api.js";
+export declare const format: (x: any) => string;
+export declare const setFormat: (width: number, prec: number) => void;
+export declare const debug: ({ data, shape, stride, offset }: ITensor<any>) => void;
+/**
+ * Syntax sugar for: `console.log(a.toString())`.
+ *
+ * @remarks
+ * Also see {@link debug} and {@link setFormat}.
+ *
+ * @param a - tensor
+ */
+export declare const print: (a: ITensor<any>) => void;
+//# sourceMappingURL=format.d.ts.map

package/format.js

@@ -0,0 +1,37 @@
+import { isNumber } from "@thi.ng/checks/is-number";
+import { floatFixedWidth } from "@thi.ng/strings/float";
+import { padLeft } from "@thi.ng/strings/pad-left";
+import { truncate } from "@thi.ng/strings/truncate";
+import { dot } from "@thi.ng/vectors/dot";
+let formatNumber;
+let pad;
+let trunc;
+const format = (x) => isNumber(x) ? formatNumber(x) : trunc(pad(x.toString()));
+const setFormat = (width, prec) => {
+  formatNumber = floatFixedWidth(width, prec);
+  pad = padLeft(width);
+  trunc = truncate(width);
+};
+setFormat(9, 4);
+const debug = ({ data, shape, stride, offset }) => {
+  let dim = shape.length;
+  const state = new Array(dim).fill(0);
+  dim--;
+  while (true) {
+    const idx = dot(state, stride) + offset;
+    console.log(state, idx, data[idx]);
+    if (++state[dim] >= shape[dim]) {
+      for (; dim-- > 0; ) if (++state[dim] < shape[dim]) break;
+      if (state[0] >= shape[0]) return;
+      state.fill(0, dim + 1);
+      dim = shape.length - 1;
+    }
+  }
+};
+const print = (a) => console.log(a.toString(), "\n");
+export {
+  debug,
+  format,
+  print,
+  setFormat
+};

package/identity.d.ts

@@ -0,0 +1,4 @@
+import type { TensorOpts, Type } from "./api.js";
+import { Tensor2 } from "./tensor.js";
+export declare const identity: <T extends Exclude<Type, "str">>(type: T, size: number, opts?: Pick<TensorOpts<any, any>, "storage">) => Tensor2<number>;
+//# sourceMappingURL=identity.d.ts.map

package/identity.js

@@ -0,0 +1,11 @@
+import { STORAGE } from "./storage.js";
+import { Tensor2 } from "./tensor.js";
+const identity = (type, size, opts) => {
+  const storage = opts?.storage || STORAGE[type];
+  const data = storage.alloc(size * size);
+  for (let i = 0; i < data.length; i += size + 1) data[i] = 1;
+  return new Tensor2(type, storage, data, [size, size], [size, 1]);
+};
+export {
+  identity
+};

package/index.d.ts

@@ -0,0 +1,60 @@
+export * from "./abs.js";
+export * from "./add.js";
+export * from "./addn.js";
+export * from "./api.js";
+export * from "./clamp.js";
+export * from "./clampn.js";
+export * from "./cos.js";
+export * from "./defopn.js";
+export * from "./defoprt.js";
+export * from "./defoprtt.js";
+export * from "./defopt.js";
+export * from "./defoptn.js";
+export * from "./defoptnn.js";
+export * from "./defoptt.js";
+export * from "./defopttt.js";
+export * from "./div.js";
+export * from "./divn.js";
+export * from "./dot.js";
+export * from "./errors.js";
+export * from "./exp.js";
+export * from "./exp2.js";
+export * from "./format.js";
+export * from "./identity.js";
+export * from "./log.js";
+export * from "./log2.js";
+export * from "./mag.js";
+export * from "./magsq.js";
+export * from "./max.js";
+export * from "./maxn.js";
+export * from "./min.js";
+export * from "./minn.js";
+export * from "./mul.js";
+export * from "./mulm.js";
+export * from "./muln.js";
+export * from "./mulv.js";
+export * from "./normalize.js";
+export * from "./pow.js";
+export * from "./pown.js";
+export * from "./product.js";
+export * from "./rand-distrib.js";
+export * from "./relu.js";
+export * from "./relun.js";
+export * from "./select.js";
+export * from "./set.js";
+export * from "./setn.js";
+export * from "./sigmoid.js";
+export * from "./sin.js";
+export * from "./softmax.js";
+export * from "./softplus.js";
+export * from "./sqrt.js";
+export * from "./step.js";
+export * from "./storage.js";
+export * from "./sub.js";
+export * from "./subn.js";
+export * from "./sum.js";
+export * from "./tan.js";
+export * from "./tanh.js";
+export * from "./tensor.js";
+export * from "./top.js";
+//# sourceMappingURL=index.d.ts.map

package/index.js

@@ -0,0 +1,59 @@
+export * from "./abs.js";
+export * from "./add.js";
+export * from "./addn.js";
+export * from "./api.js";
+export * from "./clamp.js";
+export * from "./clampn.js";
+export * from "./cos.js";
+export * from "./defopn.js";
+export * from "./defoprt.js";
+export * from "./defoprtt.js";
+export * from "./defopt.js";
+export * from "./defoptn.js";
+export * from "./defoptnn.js";
+export * from "./defoptt.js";
+export * from "./defopttt.js";
+export * from "./div.js";
+export * from "./divn.js";
+export * from "./dot.js";
+export * from "./errors.js";
+export * from "./exp.js";
+export * from "./exp2.js";
+export * from "./format.js";
+export * from "./identity.js";
+export * from "./log.js";
+export * from "./log2.js";
+export * from "./mag.js";
+export * from "./magsq.js";
+export * from "./max.js";
+export * from "./maxn.js";
+export * from "./min.js";
+export * from "./minn.js";
+export * from "./mul.js";
+export * from "./mulm.js";
+export * from "./muln.js";
+export * from "./mulv.js";
+export * from "./normalize.js";
+export * from "./pow.js";
+export * from "./pown.js";
+export * from "./product.js";
+export * from "./rand-distrib.js";
+export * from "./relu.js";
+export * from "./relun.js";
+export * from "./select.js";
+export * from "./set.js";
+export * from "./setn.js";
+export * from "./sigmoid.js";
+export * from "./sin.js";
+export * from "./softmax.js";
+export * from "./softplus.js";
+export * from "./sqrt.js";
+export * from "./step.js";
+export * from "./storage.js";
+export * from "./sub.js";
+export * from "./subn.js";
+export * from "./sum.js";
+export * from "./tan.js";
+export * from "./tanh.js";
+export * from "./tensor.js";
+export * from "./top.js";