@mni-ml/framework 0.0.1

package/dist/scalar_functions.d.ts

@@ -0,0 +1,62 @@
+import { Context } from "./autodiff.js";
+/**
+ * ScalarHistory stores how a Scalar was created.
+ * Note: inputs is typed as any[] to avoid circular dependency with Scalar
+ */
+export declare class ScalarHistory {
+    lastFn: typeof ScalarFunction | null;
+    ctx: Context | null;
+    inputs: any[];
+    constructor(lastFn?: typeof ScalarFunction | null, ctx?: Context | null, inputs?: any[]);
+}
+/**
+ * Base class for all scalar operations.
+ * Each operation implements forward (and later, backward).
+ *
+ * Note: The apply() logic lives in Scalar class to avoid circular dependencies.
+ */
+export declare abstract class ScalarFunction {
+    static forward(ctx: Context, ...inputs: number[]): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+export declare class Add extends ScalarFunction {
+    static forward(ctx: Context, a: number, b: number): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+export declare class Log extends ScalarFunction {
+    static forward(ctx: Context, a: number): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+export declare class Mul extends ScalarFunction {
+    static forward(ctx: Context, a: number, b: number): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+export declare class Inv extends ScalarFunction {
+    static forward(ctx: Context, a: number): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+export declare class Neg extends ScalarFunction {
+    static forward(ctx: Context, a: number): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+export declare class Sigmoid extends ScalarFunction {
+    static forward(ctx: Context, a: number): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+export declare class Relu extends ScalarFunction {
+    static forward(ctx: Context, a: number): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+export declare class Exp extends ScalarFunction {
+    static forward(ctx: Context, a: number): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+export declare class LT extends ScalarFunction {
+    static forward(ctx: Context, a: number, b: number): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+export declare class EQ extends ScalarFunction {
+    static forward(ctx: Context, a: number, b: number): number;
+    static backward(ctx: Context, dOut: number): number[];
+}
+//# sourceMappingURL=scalar_functions.d.ts.map

package/dist/scalar_functions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"scalar_functions.d.ts","sourceRoot":"","sources":["../src/scalar_functions.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAGxC;;;GAGG;AACH,qBAAa,aAAa;IAEX,MAAM,EAAE,OAAO,cAAc,GAAG,IAAI;IACpC,GAAG,EAAE,OAAO,GAAG,IAAI;IACnB,MAAM,EAAE,GAAG,EAAE;gBAFb,MAAM,GAAE,OAAO,cAAc,GAAG,IAAW,EAC3C,GAAG,GAAE,OAAO,GAAG,IAAW,EAC1B,MAAM,GAAE,GAAG,EAAO;CAEhC;AAED;;;;;GAKG;AACH,8BAAsB,cAAc;IAChC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM;IAIzD,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAGxD;AAED,qBAAa,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM;IAK1D,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAGxD;AAED,qBAAa,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM;IAK/C,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAIxD;AAED,qBAAa,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM;IAK1D,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAIxD;AAED,qBAAa,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM;IAK/C,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAIxD;AAED,qBAAa,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM;IAI/C,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAGxD;AAED,qBAAa,OAAQ,SAAQ,cAAc;IACvC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM;IAM/C,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAIxD;AAED,qBAAa,IAAK,SAAQ,cAAc;IACpC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM;IAK/C,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAIxD;AAED,qBAAa,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM;IAM/C,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAIxD;AAED,qBAAa,EAAG,SAAQ,cAAc;IAClC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM;IAI1D,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAGxD;AAED,qBAAa,EAAG,SAAQ,cAAc;IAClC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM;IAI1D,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE;CAGxD"}

package/dist/scalar_functions.js

@@ -0,0 +1,127 @@
+import { Context } from "./autodiff.js";
+import * as operators from "./operators.js";
+/**
+ * ScalarHistory stores how a Scalar was created.
+ * Note: inputs is typed as any[] to avoid circular dependency with Scalar
+ */
+export class ScalarHistory {
+    lastFn;
+    ctx;
+    inputs;
+    constructor(lastFn = null, ctx = null, inputs = [] // Will be Scalar[] at runtime
+    ) {
+        this.lastFn = lastFn;
+        this.ctx = ctx;
+        this.inputs = inputs;
+    }
+}
+/**
+ * Base class for all scalar operations.
+ * Each operation implements forward (and later, backward).
+ *
+ * Note: The apply() logic lives in Scalar class to avoid circular dependencies.
+ */
+export class ScalarFunction {
+    static forward(ctx, ...inputs) {
+        throw new Error("forward not implemented");
+    }
+    static backward(ctx, dOut) {
+        throw new Error("backward not implemented");
+    }
+}
+export class Add extends ScalarFunction {
+    static forward(ctx, a, b) {
+        // Don't need to save for backward since df/da = 1 and df/db = 1
+        return operators.add(a, b);
+    }
+    static backward(ctx, dOut) {
+        return [dOut, dOut];
+    }
+}
+export class Log extends ScalarFunction {
+    static forward(ctx, a) {
+        ctx.saveForBackward(a);
+        return operators.log(a);
+    }
+    static backward(ctx, dOut) {
+        const [a] = ctx.savedValues;
+        return [dOut * (1 / Math.max(a, 1e-12))];
+    }
+}
+export class Mul extends ScalarFunction {
+    static forward(ctx, a, b) {
+        ctx.saveForBackward(a, b);
+        return operators.mul(a, b);
+    }
+    static backward(ctx, dOut) {
+        const [a, b] = ctx.savedValues;
+        return [dOut * b, dOut * a];
+    }
+}
+export class Inv extends ScalarFunction {
+    static forward(ctx, a) {
+        ctx.saveForBackward(a);
+        return operators.inv(a);
+    }
+    static backward(ctx, dOut) {
+        const [a] = ctx.savedValues;
+        return [dOut * (-1 / a ** 2)];
+    }
+}
+export class Neg extends ScalarFunction {
+    static forward(ctx, a) {
+        return operators.neg(a);
+    }
+    static backward(ctx, dOut) {
+        return [dOut * (-1)];
+    }
+}
+export class Sigmoid extends ScalarFunction {
+    static forward(ctx, a) {
+        const result = operators.sigmoid(a);
+        ctx.saveForBackward(result);
+        return result;
+    }
+    static backward(ctx, dOut) {
+        const [result] = ctx.savedValues;
+        return [dOut * result * (1 - result)];
+    }
+}
+export class Relu extends ScalarFunction {
+    static forward(ctx, a) {
+        ctx.saveForBackward(a);
+        return operators.relu(a);
+    }
+    static backward(ctx, dOut) {
+        const [a] = ctx.savedValues;
+        return [dOut * (a > 0 ? 1 : 0)];
+    }
+}
+export class Exp extends ScalarFunction {
+    static forward(ctx, a) {
+        const result = operators.exp(a);
+        ctx.saveForBackward(result);
+        return result;
+    }
+    static backward(ctx, dOut) {
+        const [result] = ctx.savedValues;
+        return [dOut * result];
+    }
+}
+export class LT extends ScalarFunction {
+    static forward(ctx, a, b) {
+        return operators.lt(a, b);
+    }
+    static backward(ctx, dOut) {
+        return [0, 0];
+    }
+}
+export class EQ extends ScalarFunction {
+    static forward(ctx, a, b) {
+        return operators.eq(a, b);
+    }
+    static backward(ctx, dOut) {
+        return [0, 0];
+    }
+}
+//# sourceMappingURL=scalar_functions.js.map

package/dist/scalar_functions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"scalar_functions.js","sourceRoot":"","sources":["../src/scalar_functions.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,KAAK,SAAS,MAAM,gBAAgB,CAAC;AAE5C;;;GAGG;AACH,MAAM,OAAO,aAAa;IAEX;IACA;IACA;IAHX,YACW,SAAuC,IAAI,EAC3C,MAAsB,IAAI,EAC1B,SAAgB,EAAE,CAAE,8BAA8B;;QAFlD,WAAM,GAAN,MAAM,CAAqC;QAC3C,QAAG,GAAH,GAAG,CAAuB;QAC1B,WAAM,GAAN,MAAM,CAAY;IAC1B,CAAC;CACP;AAED;;;;;GAKG;AACH,MAAM,OAAgB,cAAc;IAChC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,GAAG,MAAgB;QAC5C,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAChD,CAAC;CACJ;AAED,MAAM,OAAO,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,CAAS,EAAE,CAAS;QAC7C,gEAAgE;QAChE,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IACxB,CAAC;CACJ;AAED,MAAM,OAAO,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,CAAS;QAClC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACvB,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IAC5B,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,MAAM,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC;QAC5B,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAE,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC;CACJ;AAED,MAAM,OAAO,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,CAAS,EAAE,CAAS;QAC7C,GAAG,CAAC,eAAe,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC1B,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/B,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC;QAC/B,OAAO,CAAC,IAAI,GAAG,CAAE,EAAE,IAAI,GAAG,CAAE,CAAC,CAAC;IAClC,CAAC;CACJ;AAED,MAAM,OAAO,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,CAAS;QAClC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACvB,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IAC5B,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,MAAM,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC;QAC5B,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,CAAE,IAAI,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC;CACJ;AAED,MAAM,OAAO,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,CAAS;QAClC,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IAC5B,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACzB,CAAC;CACJ;AAED,MAAM,OAAO,OAAQ,SAAQ,cAAc;IACvC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,CAAS;QAClC,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACpC,GAAG,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAC5B,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,MAAM,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC;QACjC,OAAO,CAAC,IAAI,GAAG,MAAO,GAAG,CAAC,CAAC,GAAG,MAAO,CAAC,CAAC,CAAC;IAC5C,CAAC;CACJ;AAED,MAAM,OAAO,IAAK,SAAQ,cAAc;IACpC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,CAAS;QAClC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACvB,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7B,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,MAAM,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC;QAC5B,OAAO,CAAC,IAAI,GAAG,CAAC,CAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC;CACJ;AAED,MAAM,OAAO,GAAI,SAAQ,cAAc;IACnC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,CAAS;QAClC,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAChC,GAAG,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAC5B,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,MAAM,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC;QACjC,OAAO,CAAC,IAAI,GAAG,MAAO,CAAC,CAAC;IAC5B,CAAC;CACJ;AAED,MAAM,OAAO,EAAG,SAAQ,cAAc;IAClC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,CAAS,EAAE,CAAS;QAC7C,OAAO,SAAS,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAClB,CAAC;CACJ;AAED,MAAM,OAAO,EAAG,SAAQ,cAAc;IAClC,MAAM,CAAC,OAAO,CAAC,GAAY,EAAE,CAAS,EAAE,CAAS;QAC7C,OAAO,SAAS,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,GAAY,EAAE,IAAY;QACtC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAClB,CAAC;CACJ"}

package/dist/tensor.d.ts

@@ -0,0 +1,58 @@
+import type { Shape } from './tensor_data.js';
+import { TensorData } from './tensor_data.js';
+import { TensorHistory, TensorFunction } from './tensor_functions.js';
+export type TensorLike = number | Tensor;
+export declare class Tensor {
+    private _data;
+    grad: Tensor | null;
+    history: TensorHistory | null;
+    constructor(data: TensorData, history?: TensorHistory | null);
+    isLeaf(): boolean;
+    requiresGrad(): boolean;
+    accumulateGrad(grad: Tensor): void;
+    chainRule(gradOutput: Tensor): [Tensor, Tensor][];
+    get parents(): Tensor[];
+    static apply(fn: typeof TensorFunction, ...vals: TensorLike[]): Tensor;
+    backward(gradOutput?: Tensor): void;
+    static tensor(values: any, shape?: Shape): Tensor;
+    static zeros(shape: Shape): Tensor;
+    static ones(shape: Shape): Tensor;
+    static rand(shape: Shape): Tensor;
+    get size(): number;
+    get dims(): number;
+    get shape(): Shape;
+    get data(): TensorData;
+    private static _ensureTensor;
+    neg(): Tensor;
+    sigmoid(): Tensor;
+    relu(): Tensor;
+    log(): Tensor;
+    exp(): Tensor;
+    inv(): Tensor;
+    add(other: number | Tensor): Tensor;
+    sub(other: number | Tensor): Tensor;
+    mul(other: number | Tensor): Tensor;
+    lt(other: number | Tensor): Tensor;
+    eq(other: number | Tensor): Tensor;
+    gt(other: number | Tensor): Tensor;
+    is_close(other: number | Tensor): Tensor;
+    radd(other: number | Tensor): Tensor;
+    rmul(other: number | Tensor): Tensor;
+    matmul(other: Tensor): Tensor;
+    conv1d(weight: Tensor): Tensor;
+    conv2d(weight: Tensor): Tensor;
+    sum(dim?: number): Tensor;
+    max(dim: number): Tensor;
+    mean(dim?: number): Tensor;
+    all(dim?: number): Tensor;
+    permute(...order: number[]): Tensor;
+    view(...shape: number[]): Tensor;
+    contiguous(): Tensor;
+    zero_grad_(): void;
+    get(idx: number[]): number;
+    set(idx: number[], value: number): void;
+    item(): number;
+    toArray(): any;
+    toString(): string;
+}
+//# sourceMappingURL=tensor.d.ts.map

package/dist/tensor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tensor.d.ts","sourceRoot":"","sources":["../src/tensor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,KAAK,EACR,MAAM,kBAAkB,CAAA;AAEzB,OAAO,EACH,UAAU,EAEb,MAAM,kBAAkB,CAAA;AAGzB,OAAO,EAEH,aAAa,EACb,cAAc,EAmBjB,MAAM,uBAAuB,CAAC;AAG/B,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG,MAAM,CAAC;AAEzC,qBAAa,MAAM;IACf,OAAO,CAAC,KAAK,CAAa;IAC1B,IAAI,EAAE,MAAM,GAAG,IAAI,CAAQ;IAC3B,OAAO,EAAE,aAAa,GAAG,IAAI,CAAQ;gBAEzB,IAAI,EAAE,UAAU,EAAE,OAAO,GAAE,aAAa,GAAG,IAAW;IAKlE,MAAM,IAAI,OAAO;IAIjB,YAAY,IAAI,OAAO;IAIvB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAQlC,SAAS,CAAC,UAAU,EAAE,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE;IAqBjD,IAAI,OAAO,IAAI,MAAM,EAAE,CAEtB;IAED,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,OAAO,cAAc,EAAE,GAAG,IAAI,EAAE,UAAU,EAAE,GAAG,MAAM;IActE,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI;IAOnC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,CAAC,EAAE,KAAK,GAAG,MAAM;IAmBjD,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAI,MAAM;IAInC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM;IAMjC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM;IASjC,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED,IAAI,KAAK,IAAI,KAAK,CAEjB;IAED,IAAI,IAAI,IAAI,UAAU,CAErB;IAED,OAAO,CAAC,MAAM,CAAC,aAAa;IAQ5B,GAAG,IAAI,MAAM;IAIb,OAAO,IAAI,MAAM;IAIjB,IAAI,IAAI,MAAM;IAId,GAAG,IAAI,MAAM;IAIb,GAAG,IAAI,MAAM;IAIb,GAAG,IAAI,MAAM;IAIb,GAAG,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM;IAInC,GAAG,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM;IAKnC,GAAG,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM;IAInC,EAAE,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM;IAIlC,EAAE,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM;IAIlC,EAAE,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM;IAMlC,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM;IAMxC,IAAI,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM;IAIpC,IAAI,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM;IAIpC,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;IAI7B,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAI9B,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAI9B,GAAG,CAAC,GAAG,CAAC,EAAE,MAAM,GAAG,MAAM;IAmBzB,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM;IAOxB,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,GAAG,MAAM;IAiB1B,GAAG,CAAC,GAAG,CAAC,EAAE,MAAM,GAAG,MAAM;IAsBzB,OAAO,CAAC,GAAG,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM;IAInC,IAAI,CAAC,GAAG,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM;IAIhC,UAAU,IAAI,MAAM;IAIpB,UAAU,IAAG,IAAI;IAIjB,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,MAAM;IAI1B,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAIvC,IAAI,IAAI,MAAM;IASd,OAAO,IAAI,GAAG;IAId,QAAQ,IAAI,MAAM;CAMrB"}

package/dist/tensor.js

@@ -0,0 +1,288 @@
+import { TensorData, shapeProduct, } from './tensor_data.js';
+import * as tensorFunctions from './tensor_functions.js';
+import { tensorMap } from './tensor_ops.js';
+import { TensorContext, TensorHistory, TensorFunction, Neg as NegFn, Sigmoid as SigmoidFn, ReLU as ReLUFn, Log as LogFn, Exp as ExpFn, Inv as InvFn, Add as AddFn, Mul as MulFn, LT as LTFn, EQ as EQFn, Sum as SumFn, Permute as PermuteFn, View as ViewFn, Contiguous as ContiguousFn, MatMul as MatMulFn, Conv1d as Conv1dFn, Conv2d as Conv2dFn, Max as MaxFn, } from './tensor_functions.js';
+import { backPropagateTensor } from './autodiff.js';
+export class Tensor {
+    _data;
+    grad = null;
+    history = null;
+    constructor(data, history = null) {
+        this._data = data;
+        this.history = history;
+    }
+    isLeaf() {
+        return !this.history?.lastFn;
+    }
+    requiresGrad() {
+        return this.history !== null;
+    }
+    accumulateGrad(grad) {
+        if (this.grad === null) {
+            this.grad = grad;
+        }
+        else {
+            this.grad = this.grad.add(grad);
+        }
+    }
+    chainRule(gradOutput) {
+        const h = this.history;
+        if (!h || !h.lastFn || !h.ctx) {
+            throw new Error("Cannot call chainRule on leaf tensor");
+        }
+        const gradients = h.lastFn.backward(h.ctx, gradOutput);
+        if (gradients.length !== h.inputs.length) {
+            throw new Error(`Backward returned ${gradients.length} gradients but expected ${h.inputs.length} for ${h.inputs.length} inputs`);
+        }
+        const result = [];
+        for (let i = 0; i < h.inputs.length; i++) {
+            result.push([h.inputs[i], gradients[i]]);
+        }
+        return result;
+    }
+    get parents() {
+        return this.history?.inputs ?? [];
+    }
+    static apply(fn, ...vals) {
+        const tensors = vals.map(v => v instanceof Tensor ? v : Tensor.tensor(v));
+        const ctx = new TensorContext();
+        const result = fn.forward(ctx, ...tensors);
+        const history = new TensorHistory(fn, ctx, tensors);
+        result.history = history;
+        return result;
+    }
+    backward(gradOutput) {
+        if (gradOutput == undefined) {
+            gradOutput = Tensor.ones(this.shape);
+        }
+        backPropagateTensor(this, gradOutput);
+    }
+    static tensor(values, shape) {
+        if (typeof values === 'number') {
+            const storage = new Float64Array([values]);
+            return new Tensor(new TensorData(storage, []));
+        }
+        const { flat, inferredShape } = flattenArray(values);
+        const finalShape = shape ?? inferredShape;
+        if (shapeProduct(finalShape) != flat.length) {
+            throw new Error(`Shape is incompatible with flat array size`);
+        }
+        const storage = new Float64Array(flat);
+        return new Tensor(new TensorData(storage, finalShape));
+    }
+    static zeros(shape) {
+        return new Tensor(TensorData.zeros(shape));
+    }
+    static ones(shape) {
+        const size = shapeProduct(shape);
+        const storage = new Float64Array(size).fill(1);
+        return new Tensor(new TensorData(storage, shape));
+    }
+    static rand(shape) {
+        const size = shapeProduct(shape);
+        const storage = new Float64Array(size);
+        for (let i = 0; i < size; i++) {
+            storage[i] = Math.random();
+        }
+        return new Tensor(new TensorData(storage, shape));
+    }
+    get size() {
+        return this._data.size;
+    }
+    get dims() {
+        return this._data.dims;
+    }
+    get shape() {
+        return this._data.shape;
+    }
+    get data() {
+        return this._data;
+    }
+    static _ensureTensor(value) {
+        if (value instanceof Tensor) {
+            return value;
+        }
+        return Tensor.tensor(value);
+    }
+    neg() {
+        return Tensor.apply(NegFn, this);
+    }
+    sigmoid() {
+        return Tensor.apply(SigmoidFn, this);
+    }
+    relu() {
+        return Tensor.apply(ReLUFn, this);
+    }
+    log() {
+        return Tensor.apply(LogFn, this);
+    }
+    exp() {
+        return Tensor.apply(ExpFn, this);
+    }
+    inv() {
+        return Tensor.apply(InvFn, this);
+    }
+    add(other) {
+        return Tensor.apply(AddFn, this, Tensor._ensureTensor(other));
+    }
+    sub(other) {
+        // a - b = a + (-b)
+        return this.add(Tensor._ensureTensor(other).neg());
+    }
+    mul(other) {
+        return Tensor.apply(MulFn, this, Tensor._ensureTensor(other));
+    }
+    lt(other) {
+        return Tensor.apply(LTFn, this, Tensor._ensureTensor(other));
+    }
+    eq(other) {
+        return Tensor.apply(EQFn, this, Tensor._ensureTensor(other));
+    }
+    gt(other) {
+        // a > b is equivalent to b < a
+        const b = Tensor._ensureTensor(other);
+        return Tensor.apply(LTFn, b, this);
+    }
+    is_close(other) {
+        // Note: is_close is a comparison operation and does not support gradients
+        const b = Tensor._ensureTensor(other);
+        return new Tensor(tensorFunctions.isClose(this._data, b._data));
+    }
+    radd(other) {
+        return this.add(other);
+    }
+    rmul(other) {
+        return this.mul(other);
+    }
+    matmul(other) {
+        return Tensor.apply(MatMulFn, this, other);
+    }
+    conv1d(weight) {
+        return Tensor.apply(Conv1dFn, this, weight);
+    }
+    conv2d(weight) {
+        return Tensor.apply(Conv2dFn, this, weight);
+    }
+    sum(dim) {
+        if (dim === undefined) {
+            if (this.dims === 0) {
+                return this;
+            }
+            let result = this;
+            for (let d = result.dims - 1; d >= 0; d--) {
+                result = Tensor.apply(SumFn(d), result);
+            }
+            return result.view();
+        }
+        if (dim < 0 || dim >= this.dims) {
+            throw new Error(`Invalid dimension ${dim} for tensor with ${this.dims} dimensions`);
+        }
+        return Tensor.apply(SumFn(dim), this);
+    }
+    max(dim) {
+        if (dim < 0 || dim >= this.dims) {
+            throw new Error(`Invalid dimension ${dim} for tensor with ${this.dims} dimensions`);
+        }
+        return Tensor.apply(MaxFn(dim), this);
+    }
+    mean(dim) {
+        if (dim === undefined) {
+            const s = this.sum();
+            const count = this.size;
+            return s.mul(1 / count);
+        }
+        if (dim < 0 || dim >= this.dims) {
+            throw new Error(`Invalid dimension ${dim} for tensor with ${this.dims} dimensions`);
+        }
+        const s = this.sum(dim);
+        const count = this._data.shape[dim];
+        return s.mul(1 / count);
+    }
+    // Note: all() is a boolean reduction and does not support gradients
+    all(dim) {
+        if (dim === undefined) {
+            let result = this._data;
+            for (let d = 0; d < result.dims; d++) {
+                result = tensorFunctions.prod(result, d);
+            }
+            const val = result.storage[0] !== 0 ? 1 : 0;
+            return Tensor.tensor(val);
+        }
+        if (dim < 0 || dim >= this.dims) {
+            throw new Error(`Invalid dimension ${dim} for tensor with ${this.dims} dimensions`);
+        }
+        const p = tensorFunctions.prod(this._data, dim);
+        const toBoolean = (x) => (x !== 0 ? 1 : 0);
+        const out = TensorData.zeros(p.shape);
+        const mapFn = tensorMap(toBoolean);
+        mapFn(out.storage, out.shape, out.strides, p.storage, p.shape, p.strides);
+        return new Tensor(out);
+    }
+    permute(...order) {
+        return Tensor.apply(PermuteFn(order), this);
+    }
+    view(...shape) {
+        return Tensor.apply(ViewFn(shape), this);
+    }
+    contiguous() {
+        return Tensor.apply(ContiguousFn, this);
+    }
+    zero_grad_() {
+        this.grad = null;
+    }
+    get(idx) {
+        return this._data.get(idx);
+    }
+    set(idx, value) {
+        this._data.set(idx, value);
+    }
+    item() {
+        if (this.size !== 1) {
+            throw new Error('item() only works for tensors with exactly one element');
+        }
+        const idx = new Array(this.dims).fill(0);
+        return this._data.get(idx);
+    }
+    toArray() {
+        return buildNestedArray(this._data, 0, new Array(this.dims).fill(0));
+    }
+    toString() {
+        if (this.dims === 0) {
+            return `Tensor(${this._data.storage[0]})`;
+        }
+        return `Tensor(${JSON.stringify(this.toArray())}, shape=[${this.shape.join(', ')}])`;
+    }
+}
+function flattenArray(arr) {
+    const shape = [];
+    let current = arr;
+    while (Array.isArray(current)) {
+        shape.push(current.length);
+        current = current[0];
+    }
+    const flat = [];
+    flattenRecursive(arr, flat);
+    return { flat, inferredShape: shape };
+}
+function flattenRecursive(arr, out) {
+    if (Array.isArray(arr)) {
+        for (const item of arr) {
+            flattenRecursive(item, out);
+        }
+    }
+    else {
+        out.push(arr);
+    }
+}
+function buildNestedArray(data, dim, idx) {
+    if (dim == data.dims) {
+        return data.get(idx);
+    }
+    const result = [];
+    for (let i = 0; i < data.shape[dim]; i++) {
+        idx[dim] = i;
+        result.push(buildNestedArray(data, dim + 1, idx));
+    }
+    return result;
+}
+//# sourceMappingURL=tensor.js.map

package/dist/tensor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tensor.js","sourceRoot":"","sources":["../src/tensor.ts"],"names":[],"mappings":"AAIA,OAAO,EACH,UAAU,EACV,YAAY,GACf,MAAM,kBAAkB,CAAA;AACzB,OAAO,KAAK,eAAe,MAAM,uBAAuB,CAAA;AACxD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAA;AAC3C,OAAO,EACH,aAAa,EACb,aAAa,EACb,cAAc,EACd,GAAG,IAAI,KAAK,EACZ,OAAO,IAAI,SAAS,EACpB,IAAI,IAAI,MAAM,EACd,GAAG,IAAI,KAAK,EACZ,GAAG,IAAI,KAAK,EACZ,GAAG,IAAI,KAAK,EACZ,GAAG,IAAI,KAAK,EACZ,GAAG,IAAI,KAAK,EACZ,EAAE,IAAI,IAAI,EACV,EAAE,IAAI,IAAI,EACV,GAAG,IAAI,KAAK,EACZ,OAAO,IAAI,SAAS,EACpB,IAAI,IAAI,MAAM,EACd,UAAU,IAAI,YAAY,EAC1B,MAAM,IAAI,QAAQ,EAClB,MAAM,IAAI,QAAQ,EAClB,MAAM,IAAI,QAAQ,EAClB,GAAG,IAAI,KAAK,GACf,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAIpD,MAAM,OAAO,MAAM;IACP,KAAK,CAAa;IAC1B,IAAI,GAAkB,IAAI,CAAC;IAC3B,OAAO,GAAyB,IAAI,CAAC;IAErC,YAAY,IAAgB,EAAE,UAAgC,IAAI;QAC9D,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IAC3B,CAAC;IAED,MAAM;QACF,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC;IACjC,CAAC;IAED,YAAY;QACR,OAAO,IAAI,CAAC,OAAO,KAAK,IAAI,CAAC;IACjC,CAAC;IAED,cAAc,CAAC,IAAY;QACvB,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;YACrB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACrB,CAAC;aAAM,CAAC;YACJ,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;IACL,CAAC;IAED,SAAS,CAAC,UAAkB;QACxB,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACvB,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,SAAS,GAAa,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QAEjE,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CACX,qBAAqB,SAAS,CAAC,MAAM,2BAA2B,CAAC,CAAC,MAAM,CAAC,MAAM,QAAQ,CAAC,CAAC,MAAM,CAAC,MAAM,SAAS,CAClH,CAAC;QACN,CAAC;QAED,MAAM,MAAM,GAAuB,EAAE,CAAC;QACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAE,EAAE,SAAS,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,IAAI,OAAO;QACP,OAAO,IAAI,CAAC,OAAO,EAAE,MAAM,IAAI,EAAE,CAAC;IACtC,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,EAAyB,EAAE,GAAG,IAAkB;QACzD,MAAM,OAAO,GAAa,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CACnC,CAAC,YAAY,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAC7C,CAAA;QAED,MAAM,GAAG,GAAG,IAAI,aAAa,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,GAAG,OAAO,CAAC;QAEzB,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,QAAQ,CAAC,UAAmB;QACxB,IAAI,UAAU,IAAI,SAAS,EAAE,CAAC;YAC1B,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzC,CAAC;QACD,mBAAmB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAC1C,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,MAAW,EAAE,KAAa;QACpC,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC7B,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YAC3C,OAAO,IAAI,MAAM,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QACrD,MAAM,UAAU,GAAG,KAAK,IAAI,aAAa,CAAC;QAE1C,IAAI,YAAY,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CACX,4CAA4C,CAC/C,CAAC;QACN,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;QACvC,OAAO,IAAI,MAAM,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,KAAY;QACrB,OAAO,IAAI,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,KAAY;QACpB,MAAM,IAAI,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QACjC,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/C,OAAO,IAAI,MAAM,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,KAAY;QACpB,MAAM,IAAI,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QACjC,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;QACvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,OAAO,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAC/B,CAAC;QACD,OAAO,IAAI,MAAM,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,IAAI,IAAI;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IAC3B,CAAC;IAED,IAAI,IAAI;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IAC3B,CAAC;IAED,IAAI,KAAK;QACL,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;IAC5B,CAAC;IAED,IAAI,IAAI;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC;IACtB,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,KAAsB;QAC/C,IAAI,KAAK,YAAY,MAAM,EAAE,CAAC;YAC1B,OAAO,KAAK,CAAC;QACjB,CAAC;QAED,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,GAAG;QACC,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACrC,CAAC;IAED,OAAO;QACH,OAAO,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IACzC,CAAC;IAED,IAAI;QACA,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IACtC,CAAC;IAED,GAAG;QACC,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACrC,CAAC;IAED,GAAG;QACC,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACrC,CAAC;IAED,GAAG;QACC,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IACrC,CAAC;IAED,GAAG,CAAC,KAAsB;QACtB,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IAClE,CAAC;IAED,GAAG,CAAC,KAAsB;QACtB,mBAAmB;QACnB,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,GAAG,CAAC,KAAsB;QACtB,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IAClE,CAAC;IAED,EAAE,CAAC,KAAsB;QACrB,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,EAAE,CAAC,KAAsB;QACrB,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,EAAE,CAAC,KAAsB;QACrB,+BAA+B;QAC/B,MAAM,CAAC,GAAG,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACtC,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;IACvC,CAAC;IAED,QAAQ,CAAC,KAAsB;QAC3B,0EAA0E;QAC1E,MAAM,CAAC,GAAG,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACtC,OAAO,IAAI,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,CAAC,KAAsB;QACvB,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAED,IAAI,CAAC,KAAsB;QACvB,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAED,MAAM,CAAC,KAAa;QAChB,OAAO,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,CAAC,MAAc;QACjB,OAAO,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,CAAC,MAAc;QACjB,OAAO,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;IAChD,CAAC;IAED,GAAG,CAAC,GAAY;QACZ,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACpB,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBAClB,OAAO,IAAI,CAAC;YAChB,CAAC;YACD,IAAI,MAAM,GAAW,IAAI,CAAC;YAC1B,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACxC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;YAC5C,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC;QACzB,CAAC;QAED,IAAI,GAAG,GAAG,CAAC,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,qBAAqB,GAAG,oBAAoB,IAAI,CAAC,IAAI,aAAa,CAAC,CAAC;QACxF,CAAC;QAED,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED,GAAG,CAAC,GAAW;QACX,IAAI,GAAG,GAAG,CAAC,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,qBAAqB,GAAG,oBAAoB,IAAI,CAAC,IAAI,aAAa,CAAC,CAAC;QACxF,CAAC;QACD,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED,IAAI,CAAC,GAAY;QACb,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACpB,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACrB,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC;YACxB,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;QAC5B,CAAC;QAED,IAAI,GAAG,GAAG,CAAC,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,qBAAqB,GAAG,oBAAoB,IAAI,CAAC,IAAI,aAAa,CAAC,CAAC;QACxF,CAAC;QAED,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAE,CAAC;QACrC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,oEAAoE;IACpE,GAAG,CAAC,GAAY;QACZ,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACpB,IAAI,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC;YACxB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBACnC,MAAM,GAAG,eAAe,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAC7C,CAAC;YACD,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC7C,OAAO,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC;QAED,IAAI,GAAG,GAAG,CAAC,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,qBAAqB,GAAG,oBAAoB,IAAI,CAAC,IAAI,aAAa,CAAC,CAAC;QACxF,CAAC;QAED,MAAM,CAAC,GAAG,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,GAAG,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QACtC,MAAM,KAAK,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC;QACnC,KAAK,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC;QAC1E,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,OAAO,CAAC,GAAG,KAAe;QACtB,OAAO,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;IAChD,CAAC;IAED,IAAI,CAAC,GAAG,KAAe;QACnB,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;IAC7C,CAAC;IAED,UAAU;QACN,OAAO,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;IAC5C,CAAC;IAED,UAAU;QACN,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACrB,CAAC;IAED,GAAG,CAAC,GAAa;QACb,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC;IAED,GAAG,CAAC,GAAa,EAAE,KAAa;QAC5B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IAC/B,CAAC;IAED,IAAI;QACA,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAC9E,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC;IAED,OAAO;QACH,OAAO,gBAAgB,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACzE,CAAC;IAED,QAAQ;QACJ,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAClB,OAAO,UAAU,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;QAC9C,CAAC;QACD,OAAO,UAAU,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,YAAY,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IACzF,CAAC;CACJ;AAED,SAAS,YAAY,CAAC,GAAQ;IAC1B,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,OAAO,GAAQ,GAAG,CAAC;IACvB,OAAO,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3B,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IACzB,CAAC;IAED,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,gBAAgB,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IAE5B,OAAO,EAAC,IAAI,EAAE,aAAa,EAAE,KAAK,EAAC,CAAC;AACxC,CAAC;AAED,SAAS,gBAAgB,CAAC,GAAQ,EAAE,GAAa;IAC7C,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACrB,KAAK,MAAM,IAAI,IAAI,GAAG,EAAE,CAAC;YACrB,gBAAgB,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QAChC,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAClB,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAgB,EAAE,GAAW,EAAE,GAAa;IAClE,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,MAAM,MAAM,GAAU,EAAE,CAAC;IACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAE,EAAE,CAAC,EAAE,EAAE,CAAC;QACxC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACb,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,GAAG,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;IACtD,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC"}

package/dist/tensor_data.d.ts

@@ -0,0 +1,29 @@
+export declare class IndexingError extends Error {
+    constructor(message: string);
+}
+export type Shape = readonly number[];
+export type Strides = readonly number[];
+export type Index = readonly number[];
+export type OutIndex = number[];
+export type Storage = Float64Array;
+export declare function indexToPosition(idx: Index, strides: Strides): number;
+export declare function toIndex(ordinal: number, shape: Shape, outIndex: OutIndex): void;
+export declare function shapeProduct(shape: Shape): number;
+export declare function strides(shape: Shape): number[];
+export declare function createSharedStorage(size: number): Storage;
+export declare class TensorData {
+    readonly storage: Storage;
+    readonly shape: Shape;
+    readonly strides: Strides;
+    readonly size: number;
+    readonly dims: number;
+    constructor(storage: Storage, shape: Shape, inputStrides?: Strides);
+    static zeros(shape: Shape): TensorData;
+    get(idx: Index): number;
+    set(idx: Index, value: number): void;
+    permute(...order: number[]): TensorData;
+    toString(): string;
+}
+export declare function shapeBroadcast(shape1: Shape, shape2: Shape): number[];
+export declare function broadcastIndex(bigIndex: Index, bigShape: Shape, shape: Shape, outIndex: OutIndex): void;
+//# sourceMappingURL=tensor_data.d.ts.map

package/dist/tensor_data.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tensor_data.d.ts","sourceRoot":"","sources":["../src/tensor_data.ts"],"names":[],"mappings":"AAAA,qBAAa,aAAc,SAAQ,KAAK;gBACxB,OAAO,EAAE,MAAM;CAI9B;AAED,MAAM,MAAM,KAAK,GAAG,SAAS,MAAM,EAAE,CAAC;AAEtC,MAAM,MAAM,OAAO,GAAG,SAAS,MAAM,EAAE,CAAC;AAExC,MAAM,MAAM,KAAK,GAAG,SAAS,MAAM,EAAE,CAAC;AAEtC,MAAM,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC;AAGhC,MAAM,MAAM,OAAO,GAAG,YAAY,CAAC;AAEnC,wBAAgB,eAAe,CAAC,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,GAAG,MAAM,CAMpE;AAED,wBAAgB,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,GAAG,IAAI,CAO/E;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,CAMjD;AAED,wBAAgB,OAAO,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,EAAE,CAQ9C;AAED,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAGzD;AAED,qBAAa,UAAU;IACnB,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAC1B,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;IACtB,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAC1B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;gBAGlB,OAAO,EAAE,OAAO,EAChB,KAAK,EAAE,KAAK,EACZ,YAAY,CAAC,EAAE,OAAO;IAe1B,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,UAAU;IAMtC,GAAG,CAAC,GAAG,EAAE,KAAK,GAAG,MAAM;IAIvB,GAAG,CAAC,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAIpC,OAAO,CAAC,GAAG,KAAK,EAAE,MAAM,EAAE,GAAG,UAAU;IA6BvC,QAAQ,IAAI,MAAM;CAGrB;AAED,wBAAgB,cAAc,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,GAAG,MAAM,EAAE,CA0BrE;AAED,wBAAgB,cAAc,CAC1B,QAAQ,EAAE,KAAK,EACf,QAAQ,EAAE,KAAK,EACf,KAAK,EAAE,KAAK,EACZ,QAAQ,EAAE,QAAQ,GACnB,IAAI,CAYN"}