numpy-ts 1.2.0 → 1.3.0

package/dist/esm/common/wasm/sort.js

@@ -1 +1 @@
-import{sort_f64 as d,sort_f32 as w,sort_f16 as x,sort_i64 as h,sort_u64 as F,sort_i32 as L,sort_u32 as P,sort_i16 as B,sort_u16 as C,sort_i8 as U,sort_u8 as M,sort_slices_f64 as R,sort_slices_f32 as I,sort_slices_f16 as k,sort_slices_i64 as N,sort_slices_u64 as O,sort_slices_i32 as Y,sort_slices_u32 as v,sort_slices_i16 as H,sort_slices_u16 as K,sort_slices_i8 as j,sort_slices_u8 as q,sort_c128 as G,sort_c64 as J,sort_slices_c128 as Q,sort_slices_c64 as V}from"./bins/sort.wasm.js";import{ensureMemory as _,resetAllocator as p,copyIn as A,copyOut as W,getSharedMemory as E,f16ToF32Input as X,f32ToF16Output as Z}from"./runtime.js";import{ArrayStorage as $}from"../storage.js";import{wasmConfig as D}from"./config.js";const z=64,T={float64:d,float32:w,float16:x,int64:h,uint64:F,int32:L,uint32:P,int16:B,uint16:C,int8:U,uint8:M,complex128:G,complex64:J},tt={float64:R,float32:I,float16:k,int64:N,uint64:O,int32:Y,uint32:v,int16:H,uint16:K,int8:j,uint8:q,complex128:Q,complex64:V},S={float64:Float64Array,float32:Float32Array,float16:Uint16Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array,complex128:Float64Array,complex64:Float32Array};function st(t,n,r,e,o){if(r<2)return!0;const a=tt[o],f=t.byteLength;if(a&&n[0]===0&&e>1&&n[1]===r){_(f),p();const s=A(t);a(s,r,e);const g=E();return new Uint8Array(t.buffer,t.byteOffset,t.byteLength).set(new Uint8Array(g.buffer,s,t.byteLength)),!0}const i=T[o],c=S[o];if(!i||!c)return!1;const b=o==="complex128"||o==="complex64",l=c.BYTES_PER_ELEMENT,y=b?l*2:l;_(f),p();const u=A(t);for(let s=0;s<e;s++)i(u+n[s]*y,r);const m=E();return new Uint8Array(t.buffer,t.byteOffset,t.byteLength).set(new Uint8Array(m.buffer,u,t.byteLength)),!0}function it(t){if(!t.isCContiguous)return null;const n=t.size;if(n<z*D.thresholdMultiplier)return null;const r=t.dtype,e=T[r],o=S[r];if(!e||!o)return null;const a=o.BYTES_PER_ELEMENT,f=n*a;_(f),p();const i=r==="float16",c=t.offset,l=r==="complex128"||r==="complex64"?n*2:n;let y=t.data.subarray(c,c+l);i&&(y=X(y,r));const u=A(y);e(u,n);const m=W(u,l,o);return $.fromData(i?Z(m,r):m,Array.from(t.shape),r)}export{it as wasmSort,st as wasmSortSlices};
+import{sort_f64 as W,sort_f32 as F,sort_f16 as M,sort_i64 as I,sort_u64 as N,sort_i32 as K,sort_u32 as O,sort_i16 as v,sort_u16 as Y,sort_i8 as H,sort_u8 as j,sort_slices_f64 as q,sort_slices_f32 as D,sort_slices_f16 as G,sort_slices_i64 as J,sort_slices_u64 as Q,sort_slices_i32 as V,sort_slices_u32 as X,sort_slices_i16 as Z,sort_slices_u16 as $,sort_slices_i8 as z,sort_slices_u8 as rr,sort_c128 as tr,sort_c64 as nr,sort_slices_c128 as er,sort_slices_c64 as or}from"./bins/sort.wasm.js";import{wasmMalloc as g,resetScratchAllocator as T,scratchAlloc as d,scratchCopyIn as C,getSharedMemory as b,f16InputToScratchF32 as sr}from"./runtime.js";import{ArrayStorage as P}from"../storage.js";import{effectiveDType as ar,hasFloat16 as U}from"../dtype.js";import{wasmConfig as h}from"./config.js";const cr=64,k={float64:W,float32:F,float16:M,int64:I,uint64:N,int32:K,uint32:O,complex128:tr,complex64:nr},R={int16:v,uint16:Y,int8:H,uint8:j},ir={float64:q,float32:D,float16:G,int64:J,uint64:Q,int32:V,uint32:X,int16:Z,uint16:$,int8:z,uint8:rr,complex128:er,complex64:or},x={float64:Float64Array,float32:Float32Array,float16:Uint16Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array,complex128:Float64Array,complex64:Float32Array};function br(r,t,e,c,o){if(e<2)return!0;const l=ir[o];if(l&&t[0]===0&&c>1&&t[1]===e){h.wasmCallCount++,T();const n=C(r);l(n,e,c);const f=b();return new Uint8Array(r.buffer,r.byteOffset,r.byteLength).set(new Uint8Array(f.buffer,n,r.byteLength)),!0}const u=k[o],y=R[o],w=x[o];if(!u&&!y||!w)return!1;const m=o==="complex128"||o==="complex64",a=w.BYTES_PER_ELEMENT,s=m?a*2:a;h.wasmCallCount++,T();const i=C(r);if(y){const n=d(e*a);for(let f=0;f<c;f++)y(i+t[f]*s,e,n)}else for(let n=0;n<c;n++)u(i+t[n]*s,e);const p=b();return new Uint8Array(r.buffer,r.byteOffset,r.byteLength).set(new Uint8Array(p.buffer,i,r.byteLength)),!0}function Ar(r){if(!r.isCContiguous)return null;const t=r.size;if(t<cr*h.thresholdMultiplier)return null;const e=ar(r.dtype),c=k[e],o=R[e],l=x[e];if(!c&&!o||!l)return null;const u=l.BYTES_PER_ELEMENT,y=e==="float16",m=e==="complex128"||e==="complex64"?t*2:t,a=m*u,s=g(a);if(!s)return null;h.wasmCallCount++,T();const i=r.offset;if(y){const n=sr(r,t);F(n,t);const f=b(),S=new Float32Array(t);S.set(new Float32Array(f.buffer,n,t)),s.release();const A=g(t*2);if(!A)return null;const E=b(),B=U?Float16Array:Uint16Array;if(U)new Float16Array(E.buffer,A.ptr,t).set(S);else{const L=new Uint16Array(E.buffer,A.ptr,t);for(let _=0;_<t;_++)L[_]=S[_]}return P.fromWasmRegion(Array.from(r.shape),e,A,t,B)}const p=b();if(r.isWasmBacked)new Uint8Array(p.buffer,s.ptr,a).set(new Uint8Array(p.buffer,r.wasmPtr+i*u,a));else{const n=r.data.subarray(i,i+m);new Uint8Array(p.buffer,s.ptr,a).set(new Uint8Array(n.buffer,n.byteOffset,n.byteLength))}if(o){const n=d(t*u);o(s.ptr,t,n)}else c(s.ptr,t);return P.fromWasmRegion(Array.from(r.shape),e,s,m,l)}export{Ar as wasmSort,br as wasmSortSlices};

package/dist/esm/common/wasm/sqrt.js

@@ -1 +1 @@
-import{sqrt_f64 as B,sqrt_f32 as g,sqrt_i64 as p,sqrt_i32 as l,sqrt_i16 as A,sqrt_i8 as c}from"./bins/sqrt.wasm.js";import{ensureMemory as m,resetAllocator as E,copyIn as T,alloc as b,copyOut as d}from"./runtime.js";import{ArrayStorage as _}from"../storage.js";import{isComplexDType as C}from"../dtype.js";import{wasmConfig as S}from"./config.js";const w=64,R={float64:B,float32:g},h={int64:p,uint64:p,int32:l,uint32:l,int16:A,uint16:A,int8:c,uint8:c},P={float64:Float64Array,float32:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array};function L(t){if(!t.isCContiguous)return null;const r=t.size;if(r<w*S.thresholdMultiplier)return null;const n=t.dtype;if(C(n))return null;const i=R[n];if(i){const y=P[n],o=y.BYTES_PER_ELEMENT;m(r*o*2),E();const e=t.data.subarray(t.offset,t.offset+r),u=T(e),a=b(r*o);i(u,a,r);const s=d(a,r,y);return _.fromData(s,Array.from(t.shape),n)}const f=h[n];if(f){const o=P[n].BYTES_PER_ELEMENT,e=8;m(r*o+r*e),E();const u=t.data.subarray(t.offset,t.offset+r),a=T(u),s=b(r*e);f(a,s,r);const D=d(s,r,Float64Array);return _.fromData(D,Array.from(t.shape),"float64")}return null}export{L as wasmSqrt};
+import{sqrt_f64 as B,sqrt_f32 as _,sqrt_i64 as T,sqrt_i32 as b,sqrt_i16_f32 as d,sqrt_i8_f32 as w}from"./bins/sqrt.wasm.js";import{wasmMalloc as f,resetScratchAllocator as y,resolveInputPtr as m,f32ToF16InPlace as g}from"./runtime.js";import{ArrayStorage as i}from"../storage.js";import{isComplexDType as P,hasFloat16 as C}from"../dtype.js";import{wasmConfig as l}from"./config.js";const R=64,S={float64:B,float32:_},h={int64:T,uint64:T,int32:b,uint32:b},I={int16:d,uint16:d,int8:w,uint8:w},p={float64:Float64Array,float32:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array};function D(r){if(!r.isCContiguous)return null;const t=r.size;if(t<R*l.thresholdMultiplier)return null;const e=r.dtype;if(P(e))return null;const c=S[e];if(c){const u=p[e],o=u.BYTES_PER_ELEMENT,a=t*o,n=f(a);if(!n)return null;l.wasmCallCount++,y();const s=m(r.data,r.isWasmBacked,r.wasmPtr,r.offset,t,o);return c(s,n.ptr,t),i.fromWasmRegion(Array.from(r.shape),e,n,t,u)}const A=I[e];if(A){const o=p[e].BYTES_PER_ELEMENT,a=t*4,n=f(a);if(!n)return null;l.wasmCallCount++,y();const s=m(r.data,r.isWasmBacked,r.wasmPtr,r.offset,t,o);return A(s,n.ptr,t),C&&(e==="int8"||e==="uint8"||e==="bool")?(g(n,t),i.fromWasmRegion(Array.from(r.shape),"float16",n,t,Float16Array)):i.fromWasmRegion(Array.from(r.shape),"float32",n,t,Float32Array)}const E=h[e];if(E){const o=p[e].BYTES_PER_ELEMENT,a=t*8,n=f(a);if(!n)return null;l.wasmCallCount++,y();const s=m(r.data,r.isWasmBacked,r.wasmPtr,r.offset,t,o);return E(s,n.ptr,t),i.fromWasmRegion(Array.from(r.shape),"float64",n,t,Float64Array)}return null}export{D as wasmSqrt};

package/dist/esm/common/wasm/square.js

@@ -1 +1 @@
-import{square_f64 as A,square_f32 as T,square_i64 as s,square_i32 as l,square_i16 as p,square_i8 as c,square_c128 as d,square_c64 as _}from"./bins/square.wasm.js";import{ensureMemory as E,resetAllocator as b,copyIn as q,alloc as g,copyOut as P}from"./runtime.js";import{ArrayStorage as S}from"../storage.js";import{isComplexDType as x}from"../dtype.js";import{wasmConfig as C}from"./config.js";const D=64,F={float64:A,float32:T,complex128:d,complex64:_,int64:s,uint64:s,int32:l,uint32:l,int16:p,uint16:p,int8:c,uint8:c},w={float64:Float64Array,float32:Float32Array,complex128:Float64Array,complex64:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array};function h(r){if(!r.isCContiguous)return null;const t=r.size;if(t<D*C.thresholdMultiplier)return null;const n=r.dtype,y=F[n],e=w[n];if(!y||!e)return null;const o=x(n)?2:1,i=e.BYTES_PER_ELEMENT,a=t*o;E(a*i*2),b();const m=q(r.data.subarray(r.offset*o,(r.offset+t)*o)),u=g(a*i);y(m,u,t);const f=P(u,a,e);return S.fromData(f,Array.from(r.shape),n)}export{h as wasmSquare};
+import{square_f64 as T,square_f32 as d,square_i64 as y,square_i32 as l,square_i16 as m,square_i8 as p,square_c128 as E,square_c64 as _}from"./bins/square.wasm.js";import{wasmMalloc as g,resetScratchAllocator as b,resolveInputPtr as q}from"./runtime.js";import{ArrayStorage as w}from"../storage.js";import{isComplexDType as C}from"../dtype.js";import{wasmConfig as c}from"./config.js";const P=64,S={float64:T,float32:d,complex128:E,complex64:_,int64:y,uint64:y,int32:l,uint32:l,int16:m,uint16:m,int8:p,uint8:p},B={float64:Float64Array,float32:Float32Array,complex128:Float64Array,complex64:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array};function k(r){if(!r.isCContiguous)return null;const n=r.size;if(n<P*c.thresholdMultiplier)return null;const t=r.dtype,i=S[t],e=B[t];if(!i||!e)return null;const s=C(t)?2:1,u=e.BYTES_PER_ELEMENT,o=n*s,f=o*u,a=g(f);if(!a)return null;c.wasmCallCount++,b();const A=q(r.data,r.isWasmBacked,r.wasmPtr,r.offset*s,o,u);return i(A,a.ptr,n),w.fromWasmRegion(Array.from(r.shape),t,a,o,e)}export{k as wasmSquare};

package/dist/esm/common/wasm/sub.js

@@ -1 +1 @@
-import{sub_f64 as z,sub_f32 as U,sub_i64 as d,sub_i32 as E,sub_i16 as T,sub_i8 as P,sub_c128 as Y,sub_c64 as v,sub_scalar_f64 as H,sub_scalar_f32 as K,sub_scalar_i64 as S,sub_scalar_i32 as B,sub_scalar_i16 as g,sub_scalar_i8 as D,sub_scalar_c128 as j,sub_scalar_c64 as q}from"./bins/sub.wasm.js";import{ensureMemory as x,resetAllocator as C,copyIn as A,alloc as w,copyOut as h}from"./runtime.js";import{ArrayStorage as F}from"../storage.js";import{promoteDTypes as G}from"../dtype.js";import{wasmConfig as R}from"./config.js";const M=64,J={float64:z,float32:U,int64:d,uint64:d,int32:E,uint32:E,int16:T,uint16:T,int8:P,uint8:P,complex128:Y,complex64:v},Q={float64:H,float32:K,int64:S,uint64:S,int32:B,uint32:B,int16:g,uint16:g,int8:D,uint8:D,complex128:j,complex64:q},O={float64:Float64Array,float32:Float32Array,complex128:Float64Array,complex64:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array},k={complex128:2,complex64:2};function tt(t,e){if(!t.isCContiguous||!e.isCContiguous)return null;const a=t.size;if(a<M*R.thresholdMultiplier)return null;const n=G(t.dtype,e.dtype),c=J[n],o=O[n];if(!c||!o)return null;const s=k[n]??1,u=o.BYTES_PER_ELEMENT,r=a*s,b=r*u,y=r*u,i=r*u;x(b+y+i),C();const l=t.offset*s,f=e.offset*s,p=t.data.subarray(l,l+r),m=e.data.subarray(f,f+r),N=A(p),I=A(m),_=w(i);c(N,I,_,a);const L=h(_,r,o);return F.fromData(L,Array.from(t.shape),n)}function rt(t,e){if(!t.isCContiguous)return null;const a=t.size;if(a<M*R.thresholdMultiplier)return null;const n=t.dtype,c=Q[n],o=O[n];if(!c||!o)return null;const s=k[n]??1,u=o.BYTES_PER_ELEMENT,r=a*s,b=r*u,y=r*u;x(b+y),C();const i=t.offset*s,l=t.data.subarray(i,i+r),f=A(l),p=w(y);c(f,p,a,e);const m=h(p,r,o);return F.fromData(m,Array.from(t.shape),n)}export{tt as wasmSub,rt as wasmSubScalar};
+import{sub_f64 as h,sub_f32 as k,sub_i64 as p,sub_i32 as b,sub_i16 as _,sub_i8 as A,sub_c128 as F,sub_c64 as M,sub_scalar_f64 as D,sub_scalar_f32 as z,sub_scalar_i64 as d,sub_scalar_i32 as E,sub_scalar_i16 as T,sub_scalar_i8 as P,sub_scalar_c128 as N,sub_scalar_c64 as I}from"./bins/sub.wasm.js";import{wasmMalloc as g,resetScratchAllocator as S,resolveInputPtr as f}from"./runtime.js";import{ArrayStorage as w}from"../storage.js";import{promoteDTypes as L}from"../dtype.js";import{wasmConfig as c}from"./config.js";const B=64,W={float64:h,float32:k,int64:p,uint64:p,int32:b,uint32:b,int16:_,uint16:_,int8:A,uint8:A,complex128:F,complex64:M},U={float64:D,float32:z,int64:d,uint64:d,int32:E,uint32:E,int16:T,uint16:T,int8:P,uint8:P,complex128:N,complex64:I},C={float64:Float64Array,float32:Float32Array,complex128:Float64Array,complex64:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array},R={complex128:2,complex64:2};function j(r,t){if(!r.isCContiguous||!t.isCContiguous||r.dtype!==t.dtype||r.size!==t.size)return null;const o=r.size;if(o<B*c.thresholdMultiplier)return null;const n=L(r.dtype,t.dtype),l=W[n],a=C[n];if(!l||!a)return null;const u=R[n]??1,i=a.BYTES_PER_ELEMENT,e=o*u,y=e*i,s=g(y);if(!s)return null;c.wasmCallCount++,S();const m=f(r.data,r.isWasmBacked,r.wasmPtr,r.offset*u,e,i),x=f(t.data,t.isWasmBacked,t.wasmPtr,t.offset*u,e,i);return l(m,x,s.ptr,o),w.fromWasmRegion(Array.from(r.shape),n,s,e,a)}function q(r,t){if(!r.isCContiguous)return null;const o=r.size;if(o<B*c.thresholdMultiplier)return null;const n=r.dtype,l=U[n],a=C[n];if(!l||!a)return null;const u=R[n]??1,i=a.BYTES_PER_ELEMENT,e=o*u,y=e*i,s=g(y);if(!s)return null;c.wasmCallCount++,S();const m=f(r.data,r.isWasmBacked,r.wasmPtr,r.offset*u,e,i);return l(m,s.ptr,o,t),w.fromWasmRegion(Array.from(r.shape),n,s,e,a)}export{j as wasmSub,q as wasmSubScalar};

package/dist/esm/common/wasm/svd.js

@@ -1 +1 @@
-import{svd_f64 as E}from"./bins/svd.wasm.js";import{ensureMemory as H,resetAllocator as O,copyIn as _,alloc as n,copyOut as i}from"./runtime.js";import{ArrayStorage as m}from"../storage.js";import{wasmConfig as v}from"./config.js";const D=4;function L(r){if(r.ndim!==2)return null;const o=r.shape[0],t=r.shape[1];if(o<D*v.thresholdMultiplier||t<D*v.thresholdMultiplier)return null;const u=Math.min(o,t),f=o*t,e=o*o,c=u,l=t*t,S=o*t+t*t,h=(f+e+c+l+S)*8;H(h),O();const y=new Float64Array(f);for(let a=0;a<o;a++)for(let s=0;s<t;s++)y[a*t+s]=Number(r.get(a,s));const F=_(y),p=n(e*8),A=n(c*8),g=n(l*8),d=n(S*8);E(F,p,A,g,d,o,t);const w=i(p,e,Float64Array),z=i(A,c,Float64Array),P=i(g,l,Float64Array),M=m.fromData(w,[o,o],"float64"),k=m.fromData(z,[u],"float64"),B=m.fromData(P,[t,t],"float64");return{u:M,s:k,vt:B}}export{L as wasmSvd};
+import{svd_f64 as W,svd_values_gk_f64 as B}from"./bins/svd.wasm.js";import{wasmMalloc as m,resetScratchAllocator as R,getSharedMemory as C}from"./runtime.js";import{ArrayStorage as S}from"../storage.js";import{isComplexDType as F}from"../dtype.js";import{wasmConfig as y}from"./config.js";const d=4;function D(e){if(e.ndim!==2||F(e.dtype))return null;const t=e.shape[0],r=e.shape[1];if(t<d*y.thresholdMultiplier||r<d*y.thresholdMultiplier)return null;const c=Math.min(t,r),f=t*t,n=c,l=r*r,o=m(f*8);if(!o)return null;const i=m(n*8);if(!i)return o.release(),null;const a=m(l*8);if(!a)return o.release(),i.release(),null;y.wasmCallCount++,R();const p=t*r,s=m(p*8);if(!s)return o.release(),i.release(),a.release(),null;const u=C();if(e.dtype==="float64"&&e.isCContiguous){const g=new Float64Array(u.buffer,s.ptr,p);e.isWasmBacked?g.set(new Float64Array(u.buffer,e.wasmPtr+e.offset*8,p)):g.set(e.data.subarray(e.offset,e.offset+p))}else{const g=new Float64Array(u.buffer,s.ptr,p);for(let w=0;w<t;w++)for(let A=0;A<r;A++)g[w*r+A]=Number(e.get(w,A))}const k=t*r+r*r,b=m(k*8);if(!b)return s.release(),o.release(),i.release(),a.release(),null;W(s.ptr,o.ptr,i.ptr,a.ptr,b.ptr,t,r),s.release(),b.release();const h=Float64Array,v=S.fromWasmRegion([t,t],"float64",o,f,h),M=S.fromWasmRegion([c],"float64",i,n,h),z=S.fromWasmRegion([r,r],"float64",a,l,h);return{u:v,s:M,vt:z}}function E(e){if(e.ndim!==2||F(e.dtype))return null;const t=e.shape[0],r=e.shape[1];if(t<d*y.thresholdMultiplier||r<d*y.thresholdMultiplier)return null;const c=Math.min(t,r),f=m(c*8);if(!f)return null;y.wasmCallCount++,R();const n=t*r,l=m(n*8);if(!l)return f.release(),null;const o=C();if(e.isCContiguous&&e.dtype==="float64"){const s=new Float64Array(o.buffer,l.ptr,n);e.isWasmBacked?s.set(new Float64Array(o.buffer,e.wasmPtr+e.offset*8,n)):s.set(e.data.subarray(e.offset,e.offset+n))}else{const s=new Float64Array(o.buffer,l.ptr,n);for(let u=0;u<n;u++)s[u]=Number(e.iget(u))}const i=t*r+4*c,a=m(i*8);if(!a)return l.release(),f.release(),null;B(l.ptr,f.ptr,a.ptr,t,r),l.release(),a.release();const p=Float64Array;return S.fromWasmRegion([c],"float64",f,c,p)}export{D as wasmSvd,E as wasmSvdValues};

package/dist/esm/common/wasm/tan.js

@@ -1 +1 @@
-import{tan_f64 as g,tan_f32 as O}from"./bins/tan.wasm.js";import{ensureMemory as p,resetAllocator as l,copyIn as c,alloc as m,copyOut as i,f16ToF32Input as B,f32ToF16Output as C}from"./runtime.js";import{ArrayStorage as b}from"../storage.js";import{isComplexDType as S,isBigIntDType as v}from"../dtype.js";import{wasmConfig as I}from"./config.js";const _=64,x={float64:g,float32:O};function R(o){if(!o.isCContiguous)return null;const r=o.size;if(r<_*I.thresholdMultiplier)return null;const e=o.dtype;if(S(e))return null;if(e==="float16"){p(r*4*2),l();const a=o.offset,u=B(o.data.subarray(a,a+r),e),n=c(u),s=m(r*4);O(n,s,r);const y=i(s,r,Float32Array);return b.fromData(C(y,e),Array.from(o.shape),e)}const A=x[e];if(A){const t=e==="float32",a=t?4:8,u=t?Float32Array:Float64Array;p(r*a*2),l();const n=o.offset,s=o.data.subarray(n,n+r),y=c(s),F=m(r*a);A(y,F,r);const k=i(F,r,u);return b.fromData(k,Array.from(o.shape),e)}const d=8;p(r*d*2),l();const T=o.offset,D=o.data,f=new Float64Array(r);if(v(e))for(let t=0;t<r;t++)f[t]=Number(D[T+t]);else for(let t=0;t<r;t++)f[t]=D[T+t];const P=c(f),w=m(r*d);g(P,w,r);const h=i(w,r,Float64Array);return b.fromData(h,Array.from(o.shape),"float64")}export{R as wasmTan};
+import{tan_f64 as B,tan_f32 as d,tan_i64_f64 as F,tan_u64_f64 as P,tan_i32_f64 as h,tan_u32_f64 as C,tan_i16_f32 as k,tan_u16_f32 as D,tan_i8_f32 as W,tan_u8_f32 as S}from"./bins/tan.wasm.js";import{wasmMalloc as i,resetScratchAllocator as m,resolveInputPtr as p,f16InputToScratchF32 as K,f32OutputToF16Region as w}from"./runtime.js";import{ArrayStorage as f}from"../storage.js";import{effectiveDType as O,isComplexDType as v,hasFloat16 as I}from"../dtype.js";import{wasmConfig as l}from"./config.js";const U=64,M={float64:B,float32:d},x={int64:F,uint64:P,int32:h,uint32:C},z={int16:k,uint16:D,int8:W,uint8:S},E={int64:8,uint64:8,int32:4,uint32:4,int16:2,uint16:2,int8:1,uint8:1};function G(r){if(!r.isCContiguous)return null;const t=r.size;if(t<U*l.thresholdMultiplier)return null;const n=O(r.dtype);if(v(n))return null;if(n==="float16"){const e=t*4,o=i(e);if(!o)return null;l.wasmCallCount++,m();const a=K(r,t);d(a,o.ptr,t);const s=w(o,t);return o.release(),s?f.fromWasmRegion(Array.from(r.shape),n,s,t,Float16Array):null}const b=M[n];if(b){const u=n==="float32",e=u?4:8,o=u?Float32Array:Float64Array,a=t*e,s=i(a);if(!s)return null;l.wasmCallCount++,m();const T=p(r.data,r.isWasmBacked,r.wasmPtr,r.offset,t,e);return b(T,s.ptr,t),f.fromWasmRegion(Array.from(r.shape),n,s,t,o)}const y=E[n];if(!y)return null;const A=z[n];if(A){const u=t*4,e=i(u);if(!e)return null;l.wasmCallCount++,m();const o=p(r.data,r.isWasmBacked,r.wasmPtr,r.offset,t,y);if(A(o,e.ptr,t),I&&(n==="int8"||n==="uint8"||n==="bool")){const a=w(e,t);return e.release(),a?f.fromWasmRegion(Array.from(r.shape),"float16",a,t,Float16Array):null}return f.fromWasmRegion(Array.from(r.shape),"float32",e,t,Float32Array)}const g=x[n];if(!g)return null;const _=t*8,c=i(_);if(!c)return null;l.wasmCallCount++,m();const R=p(r.data,r.isWasmBacked,r.wasmPtr,r.offset,t,y);return g(R,c.ptr,t),f.fromWasmRegion(Array.from(r.shape),"float64",c,t,Float64Array)}export{G as wasmTan};

package/dist/esm/common/wasm/tanh.js

@@ -1 +1 @@
-import{tanh_f64 as F,tanh_f32 as O,tanh_i64 as B,tanh_u64 as C}from"./bins/tanh.wasm.js";import{ensureMemory as u,resetAllocator as y,copyIn as c,alloc as p,copyOut as l,f16ToF32Input as S,f32ToF16Output as _}from"./runtime.js";import{ArrayStorage as m}from"../storage.js";import{isComplexDType as v,isBigIntDType as I}from"../dtype.js";import{wasmConfig as x}from"./config.js";const z=64,E={float64:F,float32:O};function K(a){if(!a.isCContiguous)return null;const r=a.size;if(r<z*x.thresholdMultiplier)return null;const o=a.dtype;if(v(o))return null;if(o==="float16"){u(r*4*2),y();const e=a.offset,f=S(a.data.subarray(e,e+r),o),n=c(f),s=p(r*4);O(n,s,r);const b=l(s,r,Float32Array);return m.fromData(_(b,o),Array.from(a.shape),o)}const A=E[o];if(A){const t=o==="float32",e=t?4:8,f=t?Float32Array:Float64Array;u(r*e*2),y();const n=a.offset,s=a.data.subarray(n,n+r),b=c(s),w=p(r*e);A(b,w,r);const k=l(w,r,f);return m.fromData(k,Array.from(a.shape),o)}if(o==="int64"||o==="uint64"){u(r*16),y();const t=a.offset,e=a.data.subarray(t,t+r),f=c(e),n=p(r*8);(o==="int64"?B:C)(f,n,r);const s=l(n,r,Float64Array);return m.fromData(s,Array.from(a.shape),"float64")}const d=8;u(r*d*2),y();const T=a.offset,D=a.data,i=new Float64Array(r);if(I(o))for(let t=0;t<r;t++)i[t]=Number(D[T+t]);else for(let t=0;t<r;t++)i[t]=D[T+t];const P=c(i),h=p(r*d);F(P,h,r);const g=l(h,r,Float64Array);return m.fromData(g,Array.from(a.shape),"float64")}export{K as wasmTanh};
+import{tanh_f64 as P,tanh_f32 as d,tanh_i64_f64 as B,tanh_u64_f64 as F,tanh_i32_f64 as R,tanh_u32_f64 as C,tanh_i16_f32 as k,tanh_u16_f32 as D,tanh_i8_f32 as W,tanh_u8_f32 as S}from"./bins/tanh.wasm.js";import{wasmMalloc as u,resetScratchAllocator as l,resolveInputPtr as p,f16InputToScratchF32 as I,f32ToF16InPlace as w}from"./runtime.js";import{ArrayStorage as s}from"../storage.js";import{effectiveDType as K,isComplexDType as v,hasFloat16 as O}from"../dtype.js";import{wasmConfig as f}from"./config.js";const U=64,M={float64:P,float32:d},x={int64:B,uint64:F,int32:R,uint32:C},z={int16:k,uint16:D,int8:W,uint8:S},E={int64:8,uint64:8,int32:4,uint32:4,int16:2,uint16:2,int8:1,uint8:1};function G(r){if(!r.isCContiguous)return null;const t=r.size;if(t<U*f.thresholdMultiplier)return null;const n=K(r.dtype);if(v(n))return null;if(n==="float16"){const e=t*4,o=u(e);if(!o)return null;f.wasmCallCount++,l();const y=I(r,t);return d(y,o.ptr,t),w(o,t),s.fromWasmRegion(Array.from(r.shape),n,o,t,Float16Array)}const b=M[n];if(b){const a=n==="float32",e=a?4:8,o=a?Float32Array:Float64Array,y=t*e,c=u(y);if(!c)return null;f.wasmCallCount++,l();const T=p(r.data,r.isWasmBacked,r.wasmPtr,r.offset,t,e);return b(T,c.ptr,t),s.fromWasmRegion(Array.from(r.shape),n,c,t,o)}const i=E[n];if(!i)return null;const A=z[n];if(A){const a=t*4,e=u(a);if(!e)return null;f.wasmCallCount++,l();const o=p(r.data,r.isWasmBacked,r.wasmPtr,r.offset,t,i);return A(o,e.ptr,t),O&&(n==="int8"||n==="uint8"||n==="bool")?(w(e,t),s.fromWasmRegion(Array.from(r.shape),"float16",e,t,Float16Array)):s.fromWasmRegion(Array.from(r.shape),"float32",e,t,Float32Array)}const h=x[n];if(!h)return null;const g=t*8,m=u(g);if(!m)return null;f.wasmCallCount++,l();const _=p(r.data,r.isWasmBacked,r.wasmPtr,r.offset,t,i);return h(_,m.ptr,t),s.fromWasmRegion(Array.from(r.shape),"float64",m,t,Float64Array)}export{G as wasmTanh};

package/dist/esm/common/wasm/tile.js

@@ -1 +1 @@
-import{tile_2d_f64 as F,tile_2d_f32 as T,tile_2d_i64 as _,tile_2d_i32 as b,tile_2d_i16 as E,tile_2d_i8 as D}from"./bins/tile.wasm.js";import{ensureMemory as P,resetAllocator as S,copyIn as h,alloc as w,copyOut as I,f16ToF32Input as C,f32ToF16Output as M}from"./runtime.js";import{ArrayStorage as k}from"../storage.js";import{wasmConfig as O}from"./config.js";const R=64,U={float64:F,float32:T,int64:_,uint64:_,int32:b,uint32:b,int16:E,uint16:E,int8:D,uint8:D,float16:T},z={float64:Float64Array,float32:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array,float16:Float32Array};function v(t,n,o){if(!t.isCContiguous||t.shape.length!==2)return null;const e=t.size;if(e<R*O.thresholdMultiplier)return null;const r=t.dtype,y=U[r],a=z[r];if(!y||!a)return null;const i=t.shape[0],u=t.shape[1],s=i*n*u*o,f=a.BYTES_PER_ELEMENT,g=e*f,p=s*f;P(g+p),S();const c=r==="float16",m=t.offset;let l=t.data.subarray(m,m+e);c&&(l=C(l,r));const B=h(l),A=w(p);y(B,A,i,u,n,o);const d=I(A,s,a);return k.fromData(c?M(d,r):d,[i*n,u*o],r)}export{v as wasmTile2D};
+import{tile_2d_f64 as h,tile_2d_f32 as c,tile_2d_i64 as A,tile_2d_i32 as p,tile_2d_i16 as y,tile_2d_i8 as d}from"./bins/tile.wasm.js";import{wasmMalloc as B,resetScratchAllocator as S,resolveInputPtr as b}from"./runtime.js";import{ArrayStorage as T}from"../storage.js";import{effectiveDType as D}from"../dtype.js";import{wasmConfig as _}from"./config.js";const k=64,F={float64:h,float32:c,int64:A,uint64:A,int32:p,uint32:p,int16:y,uint16:y,int8:d,uint8:d,float16:c},C={float64:Float64Array,float32:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array,float16:Float32Array};function z(t,r,n){if(!t.isCContiguous||t.shape.length!==2)return null;const u=t.size;if(u<k*_.thresholdMultiplier)return null;const e=D(t.dtype),f=F[e],l=C[e];if(!f||!l)return null;const o=t.shape[0],i=t.shape[1],s=o*r*i*n,m=l.BYTES_PER_ELEMENT,g=s*m,E=e==="float16",a=B(g);if(!a)return null;if(_.wasmCallCount++,S(),E){const P=b(t.data,t.isWasmBacked,t.wasmPtr,t.offset,u,2);return y(P,a.ptr,o,i,r,n),T.fromWasmRegion([o*r,i*n],e,a,s,Float16Array)}const w=b(t.data,t.isWasmBacked,t.wasmPtr,t.offset,u,m);return f(w,a.ptr,o,i,r,n),T.fromWasmRegion([o*r,i*n],e,a,s,l)}export{z as wasmTile2D};

package/dist/esm/common/wasm/unravel_index.js

@@ -0,0 +1 @@
+import{unravel_index_i32_f64 as B,unravel_index_i64_f64 as S,unravel_index_f64 as T}from"./bins/unravel_index.wasm.js";import{wasmMalloc as _,resetScratchAllocator as x,resolveInputPtr as P,scratchAlloc as C,scratchCopyIn as i,getSharedMemory as W}from"./runtime.js";import{ArrayStorage as v}from"../storage.js";import{wasmConfig as h}from"./config.js";const R=64;function H(o,f){if(!o.isCContiguous)return null;const t=o.size,n=f.length;if(n===0||t===0||t<R*h.thresholdMultiplier)return null;const c=o.dtype,p=o.data,d=o.offset,A=8,w=[];for(let r=0;r<n;r++){const s=_(t*A);if(!s){for(const a of w)a.release();return null}w.push(s)}h.wasmCallCount++,x();const g=C(n*t*A);if(c==="int64"||c==="uint64"){const s=P(p,o.isWasmBacked,o.wasmPtr,d,t,8),a=new BigInt64Array(n);let u=1n;for(let e=n-1;e>=0;e--)a[e]=u,u*=BigInt(f[e]);const y=i(a),m=new BigInt64Array(f.map(BigInt)),l=i(m);S(s,g,t,y,l,n)}else if(c==="int32"||c==="uint32"){const s=P(p,o.isWasmBacked,o.wasmPtr,d,t,4),a=new Int32Array(n);let u=1;for(let e=n-1;e>=0;e--)a[e]=u,u*=f[e];const y=i(a),m=new Int32Array(f),l=i(m);B(s,g,t,y,l,n)}else{let r;if(c==="float64")r=P(p,o.isWasmBacked,o.wasmPtr,d,t,A);else{const l=new Float64Array(t);for(let e=0;e<t;e++)l[e]=Number(p[d+e]);r=i(l)}const s=new Int32Array(n);let a=1;for(let l=n-1;l>=0;l--)s[l]=a,a*=f[l];const u=i(s),y=new Int32Array(f),m=i(y);T(r,g,t,u,m,n)}const I=W();for(let r=0;r<n;r++){const s=new Float64Array(I.buffer,g+r*t*A,t);new Float64Array(I.buffer,w[r].ptr,t).set(s)}const k=Array.from(o.shape),b=[];for(let r=0;r<n;r++)b.push(v.fromWasmRegion(k,"float64",w[r],t,Float64Array));return b}export{H as wasmUnravelIndex};

package/dist/esm/common/wasm/vdot.js

@@ -1 +1 @@
-import{vdot_c128 as b,vdot_c64 as A}from"./bins/vdot.wasm.js";import{ensureMemory as E,resetAllocator as C,copyIn as p,alloc as D,copyOut as g}from"./runtime.js";import{promoteDTypes as w}from"../dtype.js";import{Complex as x}from"../complex.js";import{wasmConfig as P}from"./config.js";const B=64,S={complex128:b,complex64:A},_={complex128:Float64Array,complex64:Float32Array};function K(r,t){if(r.ndim!==1||t.ndim!==1||!r.isCContiguous||!t.isCContiguous)return null;const e=r.shape[0];if(e!==t.shape[0]||e<B*P.thresholdMultiplier)return null;const s=w(r.dtype,t.dtype),a=S[s],o=_[s];if(!a||!o)return null;const n=o.BYTES_PER_ELEMENT,y=e*2*n,c=e*2*n,u=2*n;E(y+c+u),C();const f=r.data.subarray(r.offset*2,r.offset*2+e*2),i=t.data.subarray(t.offset*2,t.offset*2+e*2),d=p(f),T=p(i),m=D(u);a(d,T,m,e);const l=g(m,2,o);return new x(Number(l[0]),Number(l[1]))}export{K as wasmVdotComplex};
+import{vdot_c128 as d,vdot_c64 as w}from"./bins/vdot.wasm.js";import{resetScratchAllocator as E,resolveInputPtr as l,scratchAlloc as A,getSharedMemory as T}from"./runtime.js";import{promoteDTypes as C}from"../dtype.js";import{Complex as P}from"../complex.js";import{wasmConfig as p}from"./config.js";const b=64,g={complex128:d,complex64:w},h={complex128:Float64Array,complex64:Float32Array};function R(r,t){if(r.ndim!==1||t.ndim!==1||!r.isCContiguous||!t.isCContiguous)return null;const e=r.shape[0];if(e!==t.shape[0]||e<b*p.thresholdMultiplier)return null;const s=C(r.dtype,t.dtype),m=g[s],o=h[s];if(!m||!o)return null;const n=o.BYTES_PER_ELEMENT,c=2*n;p.wasmCallCount++,E();const i=l(r.data,r.isWasmBacked,r.wasmPtr,r.offset*2,e*2,n),y=l(t.data,t.isWasmBacked,t.wasmPtr,t.offset*2,e*2,n),a=A(c);m(i,y,a,e);const f=T(),u=new o(f.buffer,a,2);return new P(Number(u[0]),Number(u[1]))}export{R as wasmVdotComplex};

package/dist/esm/common/wasm/vecdot.js

@@ -1 +1 @@
-import{vecdot_f64 as P,vecdot_f32 as d,vecdot_c128 as h,vecdot_c64 as x,vecdot_i64 as A,vecdot_i32 as T,vecdot_i16 as E,vecdot_i8 as _}from"./bins/vecdot.wasm.js";import{ensureMemory as C,resetAllocator as S,copyIn as b,alloc as w,copyOut as I,f16ToF32Input as g,f32ToF16Output as R}from"./runtime.js";import{ArrayStorage as M}from"../storage.js";import{promoteDTypes as O}from"../dtype.js";import{wasmConfig as U}from"./config.js";const k=256,K={float64:P,float32:d,complex128:h,complex64:x,int64:A,uint64:A,int32:T,uint32:T,int16:E,uint16:E,int8:_,uint8:_,float16:d},L={float64:Float64Array,float32:Float32Array,complex128:Float64Array,complex64:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array,float16:Float32Array},V={complex128:2,complex64:2};function q(o,e){if(o.ndim!==2||e.ndim!==2||!o.isCContiguous||!e.isCContiguous)return null;const t=o.shape[0],a=o.shape[1];if(t!==e.shape[0]||a!==e.shape[1]||t*a<k*U.thresholdMultiplier)return null;const n=O(o.dtype,e.dtype),c=K[n],s=L[n];if(!c||!s)return null;const r=V[n]??1,y=s.BYTES_PER_ELEMENT,B=t*a*r*y,D=t*a*r*y,l=t*r*y;C(B+D+l),S();const p=n==="float16";let i=o.data.subarray(o.offset*r,o.offset*r+t*a*r),u=e.data.subarray(e.offset*r,e.offset*r+t*a*r);p&&(i=g(i,n),u=g(u,n));const F=b(i),v=b(u),f=w(l);c(F,v,f,t,a);const m=I(f,t*r,s);return M.fromData(p?R(m,n):m,[t],n)}export{q as wasmVecdot};
+import*as b from"./bins/vecdot_float.wasm.js";import*as h from"./bins/vecdot_float-relaxed.wasm.js";import{vecdot_i64 as y,vecdot_i32 as p,vecdot_i16 as d,vecdot_i8 as A}from"./bins/vecdot_int.wasm.js";import{useRelaxedKernels as x}from"./detect.js";import{wasmMalloc as C,resetScratchAllocator as S,resolveInputPtr as T,f16InputToScratchF32 as g,f32OutputToF16Region as D}from"./runtime.js";import{ArrayStorage as B}from"../storage.js";import{effectiveDType as k,promoteDTypes as I}from"../dtype.js";import{wasmConfig as _}from"./config.js";let W=null;function s(){return W??=x()?h:b}const M=256,K={float64:(...t)=>s().vecdot_f64(...t),float32:(...t)=>s().vecdot_f32(...t),complex128:(...t)=>s().vecdot_c128(...t),complex64:(...t)=>s().vecdot_c64(...t),int64:y,uint64:y,int32:p,uint32:p,int16:d,uint16:d,int8:A,uint8:A,float16:(...t)=>s().vecdot_f32(...t)},U={float64:Float64Array,float32:Float32Array,complex128:Float64Array,complex64:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array,float16:Float32Array},L={complex128:2,complex64:2};function q(t,e){if(t.ndim!==2||e.ndim!==2||!t.isCContiguous||!e.isCContiguous)return null;const r=t.shape[0],o=t.shape[1];if(r!==e.shape[0]||o!==e.shape[1]||r*o<M*_.thresholdMultiplier)return null;const n=k(I(t.dtype,e.dtype)),u=K[n],f=U[n];if(!u||!f)return null;const l=L[n]??1,c=f.BYTES_PER_ELEMENT,i=r*l,E=i*c,w=n==="float16",a=C(E);if(!a)return null;if(_.wasmCallCount++,S(),w){const v=g(t,r*o),F=g(e,r*o);u(v,F,a.ptr,r,o);const m=D(a,i);return a.release(),m?B.fromWasmRegion([r],n,m,i,Float16Array):null}const P=T(t.data,t.isWasmBacked,t.wasmPtr,t.offset*l,r*o*l,c),R=T(e.data,e.isWasmBacked,e.wasmPtr,e.offset*l,r*o*l,c);return u(P,R,a.ptr,r,o),B.fromWasmRegion([r],n,a,i,f)}export{q as wasmVecdot};

package/dist/esm/common/wasm/vecmat.js

@@ -1 +1 @@
-import{vecmat_f64 as P,vecmat_f32 as A,vecmat_c128 as C,vecmat_c64 as S,vecmat_i64 as T,vecmat_i32 as d,vecmat_i16 as b,vecmat_i8 as E}from"./bins/vecmat.wasm.js";import{ensureMemory as h,resetAllocator as w,copyIn as _,alloc as x,copyOut as I,f16ToF32Input as g,f32ToF16Output as R}from"./runtime.js";import{ArrayStorage as M}from"../storage.js";import{promoteDTypes as N}from"../dtype.js";import{wasmConfig as O}from"./config.js";const U=128,k={float64:P,float32:A,complex128:C,complex64:S,int64:T,uint64:T,int32:d,uint32:d,int16:b,uint16:b,int8:E,uint8:E,float16:A},K={float64:Float64Array,float32:Float32Array,complex128:Float64Array,complex64:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array,float16:Float32Array},L={complex128:2,complex64:2};function q(a,r){if(a.ndim!==1||r.ndim!==2||!a.isCContiguous||!r.isCContiguous)return null;const n=r.shape[0],o=r.shape[1];if(n!==a.shape[0]||n*o<U*O.thresholdMultiplier)return null;const e=N(a.dtype,r.dtype),c=k[e],y=K[e];if(!c||!y)return null;const t=L[e]??1,s=y.BYTES_PER_ELEMENT,D=n*t*s,F=n*o*t*s,l=o*t*s;h(D+F+l),w();const m=e==="float16";let i=a.data.subarray(a.offset*t,a.offset*t+n*t),u=r.data.subarray(r.offset*t,r.offset*t+n*o*t);m&&(i=g(i,e),u=g(u,e));const v=_(i),B=_(u),p=x(l);c(v,B,p,n,o);const f=I(p,o*t,y);return M.fromData(m?R(f,e):f,[o],e)}export{q as wasmVecmat};
+import*as P from"./bins/vecmat_float.wasm.js";import*as C from"./bins/vecmat_float-relaxed.wasm.js";import{vecmat_i64 as y,vecmat_i32 as p,vecmat_i16 as A,vecmat_i8 as d}from"./bins/vecmat_int.wasm.js";import{useRelaxedKernels as S}from"./detect.js";import{wasmMalloc as h,resetScratchAllocator as D,resolveInputPtr as T,f16InputToScratchF32 as b,f32OutputToF16Region as k}from"./runtime.js";import{ArrayStorage as g}from"../storage.js";import{effectiveDType as I,promoteDTypes as W}from"../dtype.js";import{wasmConfig as _}from"./config.js";let x=null;function s(){return x??=S()?C:P}const M=128,K={float64:(...t)=>s().vecmat_f64(...t),float32:(...t)=>s().vecmat_f32(...t),complex128:(...t)=>s().vecmat_c128(...t),complex64:(...t)=>s().vecmat_c64(...t),int64:y,uint64:y,int32:p,uint32:p,int16:A,uint16:A,int8:d,uint8:d,float16:(...t)=>s().vecmat_f32(...t)},N={float64:Float64Array,float32:Float32Array,complex128:Float64Array,complex64:Float32Array,int64:BigInt64Array,uint64:BigUint64Array,int32:Int32Array,uint32:Uint32Array,int16:Int16Array,uint16:Uint16Array,int8:Int8Array,uint8:Uint8Array,float16:Float32Array},U={complex128:2,complex64:2};function q(t,r){if(t.ndim!==1||r.ndim!==2||!t.isCContiguous||!r.isCContiguous)return null;const e=r.shape[0],n=r.shape[1];if(e!==t.shape[0]||e*n<M*_.thresholdMultiplier)return null;const a=I(W(t.dtype,r.dtype)),m=K[a],u=N[a];if(!m||!u)return null;const l=U[a]??1,f=u.BYTES_PER_ELEMENT,i=n*l,w=i*f,B=a==="float16",o=h(w);if(!o)return null;if(_.wasmCallCount++,D(),B){const v=b(t,e),F=b(r,e*n);m(v,F,o.ptr,e,n);const c=k(o,i);return o.release(),c?g.fromWasmRegion([n],a,c,i,Float16Array):null}const E=T(t.data,t.isWasmBacked,t.wasmPtr,t.offset*l,e*l,f),R=T(r.data,r.isWasmBacked,r.wasmPtr,r.offset*l,e*n*l,f);return m(E,R,o.ptr,e,n),g.fromWasmRegion([n],a,o,i,u)}export{q as wasmVecmat};

package/dist/esm/common/wasm/vector_norm.js

@@ -0,0 +1 @@
+import*as a from"./bins/vector_norm.wasm.js";import*as s from"./bins/vector_norm-relaxed.wasm.js";import{useRelaxedKernels as u}from"./detect.js";import{resetScratchAllocator as p,resolveInputPtr as i,f16InputToScratchF32 as c}from"./runtime.js";import{effectiveDType as y,isComplexDType as b}from"../dtype.js";import{wasmConfig as o}from"./config.js";let d=null;function n(){return d??=u()?s:a}const C=64,_={float64:{fn:(...t)=>n().vector_norm2_f64(...t),bpe:8},float32:{fn:(...t)=>n().vector_norm2_f32(...t),bpe:4}};function D(t){if(!t.isCContiguous||b(t.dtype))return null;const r=t.size;if(r<C*o.thresholdMultiplier)return null;const l=y(t.dtype);if(o.wasmCallCount++,p(),l==="float16"){const m=c(t,r);return n().vector_norm2_f32(m,r)}const e=_[l];if(!e)return o.wasmCallCount--,null;const f=i(t.data,t.isWasmBacked,t.wasmPtr,t.offset,r,e.bpe);return e.fn(f,r)}export{D as wasmVectorNorm2};

package/dist/esm/core/advanced.js

@@ -1 +1 @@
-import{NDArrayCore as D}from"../common/ndarray-core.js";import*as a from"../common/ops/advanced.js";import*as N from"../common/ops/comparison.js";import{array as b}from"./creation.js";import{toStorage as n,fromStorage as i,fromStorageView as C}from"./types.js";import{computeBroadcastShape as _}from"../common/broadcasting.js";import{isBigIntDType as d}from"../common/dtype.js";function q(e,r){return C(a.broadcast_to(n(e),r),e)}function E(...e){const r=e.map(n);return a.broadcast_arrays(r).map(o=>i(o))}function B(...e){if(e.length===0)return[];if(e.length===1)return[...e[0]];let r=[...e[0]];for(let o=1;o<e.length;o++){const t=e[o],s=Math.max(r.length,t.length),m=[];for(let c=0;c<s;c++){const y=c<r.length?r[r.length-1-c]:1,f=c<t.length?t[t.length-1-c]:1;if(y===f)m.unshift(y);else if(y===1)m.unshift(f);else if(f===1)m.unshift(y);else throw new Error(`Cannot broadcast shapes: dimensions ${y} and ${f} are incompatible`)}r=m}return r}function x(e,r,o){return i(a.take(n(e),r,o))}function F(e,r,o){const t=Array.isArray(o)?n(b(o)):n(o);a.put(n(e),r,t)}function M(e,r,o){return i(a.take_along_axis(n(e),n(r),o))}function T(e,r,o,t){a.put_along_axis(n(e),n(r),n(o),t)}function z(e,r){const o=r.map(n);return i(a.choose(n(e),o))}function g(e,r,o){return i(a.compress(n(e),n(r),o))}function I(e,r,o=0){let t;return r instanceof D?t=Array.from(r.data):Array.isArray(r[0])?t=r.flat():t=r,x(e,t,o)}function O(e,r,o){return g(r,e,o)}function V(e,r,o=0){const t=e.map(n),s=r.map(n);return i(a.select(t,s,o))}function L(e,r,o){a.place(n(e),n(r),n(o))}function R(e,r,o){a.putmask(n(e),n(r),n(o))}function G(e,r){const o=n(e),t=e.shape,s=e.size,m=e.dtype;if(typeof r=="number"||typeof r=="bigint"){e.fill(r);return}const c=n(r),y=r.shape,f=_([y,t]);if(!f)throw new Error(`could not broadcast input array from shape (${y.join(",")}) into shape (${t.join(",")})`);if(f.length!==t.length||!f.every((u,p)=>u===t[p]))throw new Error(`could not broadcast input array from shape (${y.join(",")}) into shape (${t.join(",")})`);const A=a.broadcast_to(c,t);if(d(m))for(let u=0;u<s;u++){const p=A.iget(u),l=typeof p=="bigint"?p:BigInt(Math.round(Number(p)));o.iset(u,l)}else if(m==="bool")for(let u=0;u<s;u++){const p=A.iget(u);o.iset(u,p?1:0)}else for(let u=0;u<s;u++){const p=A.iget(u);o.iset(u,Number(p))}}function H(e,r="int32"){return i(a.indices(e,r))}function J(...e){const r=e.map(n);return a.ix_(...r).map(o=>i(o))}function K(e,r,o="raise"){const t=e.map(n);return i(a.ravel_multi_index(t,r,o))}function P(e,r){const o=typeof e=="number"?n(b([e])):n(e);return a.unravel_index(o,r).map(t=>i(t))}function Q(e,r=2){return a.diag_indices(e,r).map(o=>i(o))}function U(e){return a.diag_indices_from(n(e)).map(r=>i(r))}function W(e,r,o=!1){a.fill_diagonal(n(e),r,o)}function X(e,r=0,o){return a.tril_indices(e,r,o).map(t=>i(t))}function Y(e,r=0){return a.tril_indices_from(n(e),r).map(o=>i(o))}function Z(e,r=0,o){return a.triu_indices(e,r,o).map(t=>i(t))}function rr(e,r=0){return a.triu_indices_from(n(e),r).map(o=>i(o))}function er(e,r,o=0){const t=(s,m)=>n(r(s,m));return a.mask_indices(e,t,o).map(s=>i(s))}function or(e,r,o=!1){return a.array_equal(n(e),n(r),o)}function nr(e,r){return N.arrayEquiv(n(e),n(r))}function tr(e,r,o){const t=s=>{const m=e(i(s));return typeof m=="number"?m:n(m)};return i(a.apply_along_axis(n(o),r,t))}function ar(e,r,o){const t=(s,m)=>n(e(i(s),m));return i(a.apply_over_axes(n(r),t,o))}function ir(e,r){return a.may_share_memory(n(e),n(r))}function ur(e,r){return a.shares_memory(n(e),n(r))}const sr=a.geterr,mr=a.seterr;export{tr as apply_along_axis,ar as apply_over_axes,or as array_equal,nr as array_equiv,O as bindex,E as broadcast_arrays,B as broadcast_shapes,q as broadcast_to,z as choose,g as compress,G as copyto,Q as diag_indices,U as diag_indices_from,W as fill_diagonal,sr as geterr,I as iindex,H as indices,J as ix_,er as mask_indices,ir as may_share_memory,L as place,F as put,T as put_along_axis,R as putmask,K as ravel_multi_index,V as select,mr as seterr,ur as shares_memory,x as take,M as take_along_axis,X as tril_indices,Y as tril_indices_from,Z as triu_indices,rr as triu_indices_from,P as unravel_index};
+import{NDArrayCore as b}from"../common/ndarray-core.js";import*as t from"../common/ops/advanced.js";import*as N from"../common/ops/comparison.js";import{array as D}from"./creation.js";import{toStorage as n,fromStorage as i,fromStorageView as C}from"./types.js";import{computeBroadcastShape as d}from"../common/broadcasting.js";import{isBigIntDType as _}from"../common/dtype.js";function q(e,r){return C(t.broadcast_to(n(e),r),e)}function E(...e){const r=e.map(n);return t.broadcast_arrays(r).map(o=>i(o))}function I(...e){if(e.length===0)return[];if(e.length===1)return[...e[0]];let r=[...e[0]];for(let o=1;o<e.length;o++){const a=e[o],s=Math.max(r.length,a.length),m=[];for(let c=0;c<s;c++){const y=c<r.length?r[r.length-1-c]:1,A=c<a.length?a[a.length-1-c]:1;if(y===A)m.unshift(y);else if(y===1)m.unshift(A);else if(A===1)m.unshift(y);else throw new Error(`Cannot broadcast shapes: dimensions ${y} and ${A} are incompatible`)}r=m}return r}function g(e,r,o){return i(t.take(n(e),r,o))}function B(e,r,o){const a=Array.isArray(o)?n(D(o)):n(o);t.put(n(e),r,a)}function F(e,r,o){return i(t.take_along_axis(n(e),n(r),o))}function M(e,r,o,a){t.put_along_axis(n(e),n(r),n(o),a)}function T(e,r){const o=r.map(n);return i(t.choose(n(e),o))}function x(e,r,o){return i(t.compress(n(e),n(r),o))}function z(e,r,o=0){let a;return r instanceof b?a=Array.from(r.data):Array.isArray(r[0])?a=r.flat():a=r,g(e,a,o)}function O(e,r,o){return x(r,e,o)}function V(e,r,o=0){const a=e.map(n),s=r.map(n);return i(t.select(a,s,o))}function L(e,r,o){t.place(n(e),n(r),n(o))}function R(e,r,o){t.putmask(n(e),n(r),n(o))}function G(e,r){const o=n(e),a=e.shape,s=e.size,m=e.dtype;if(typeof r=="number"||typeof r=="bigint"){e.fill(r);return}const c=n(r),y=r.shape,A=d([y,a]);if(!A)throw new Error(`could not broadcast input array from shape (${y.join(",")}) into shape (${a.join(",")})`);if(A.length!==a.length||!A.every((u,p)=>u===a[p]))throw new Error(`could not broadcast input array from shape (${y.join(",")}) into shape (${a.join(",")})`);const f=t.broadcast_to(c,a);if(_(m))for(let u=0;u<s;u++){const p=f.iget(u),l=typeof p=="bigint"?p:BigInt(Math.round(Number(p)));o.iset(u,l)}else if(m==="bool")for(let u=0;u<s;u++){const p=f.iget(u);o.iset(u,p?1:0)}else for(let u=0;u<s;u++){const p=f.iget(u);o.iset(u,p)}}function H(e,r="float64"){return i(t.indices(e,r))}function J(...e){const r=e.map(n);return t.ix_(...r).map(o=>i(o))}function K(e,r,o="raise"){const a=e.map(n);return i(t.ravel_multi_index(a,r,o))}function P(e,r){const o=typeof e=="number"?n(D([e])):n(e);return t.unravel_index(o,r).map(a=>i(a))}function Q(e,r=2){return t.diag_indices(e,r).map(o=>i(o))}function U(e){return t.diag_indices_from(n(e)).map(r=>i(r))}function W(e,r,o=!1){t.fill_diagonal(n(e),r,o)}function X(e,r=0,o){return t.tril_indices(e,r,o).map(a=>i(a))}function Y(e,r=0){return t.tril_indices_from(n(e),r).map(o=>i(o))}function Z(e,r=0,o){return t.triu_indices(e,r,o).map(a=>i(a))}function rr(e,r=0){return t.triu_indices_from(n(e),r).map(o=>i(o))}function er(e,r,o=0){const a=(s,m)=>n(r(i(s),m));return t.mask_indices(e,a,o).map(s=>i(s))}function or(e,r,o=!1){return t.array_equal(n(e),n(r),o)}function nr(e,r){return N.arrayEquiv(n(e),n(r))}function ar(e,r,o){const a=s=>{const m=e(i(s));return typeof m=="number"?m:n(m)};return i(t.apply_along_axis(n(o),r,a))}function tr(e,r,o){const a=(s,m)=>n(e(i(s),m));return i(t.apply_over_axes(n(r),a,o))}function ir(e,r){return t.may_share_memory(n(e),n(r))}function ur(e,r){return t.shares_memory(n(e),n(r))}const sr=t.geterr,mr=t.seterr;function yr(e,...r){const o=r.map(a=>a instanceof b?n(a):a);return i(t.vindex(n(e),...o))}export{ar as apply_along_axis,tr as apply_over_axes,or as array_equal,nr as array_equiv,O as bindex,E as broadcast_arrays,I as broadcast_shapes,q as broadcast_to,T as choose,x as compress,G as copyto,Q as diag_indices,U as diag_indices_from,W as fill_diagonal,sr as geterr,z as iindex,H as indices,J as ix_,er as mask_indices,ir as may_share_memory,L as place,B as put,M as put_along_axis,R as putmask,K as ravel_multi_index,V as select,mr as seterr,ur as shares_memory,g as take,F as take_along_axis,X as tril_indices,Y as tril_indices_from,Z as triu_indices,rr as triu_indices_from,P as unravel_index,yr as vindex};

package/dist/esm/core/creation.js

@@ -1 +1 @@
-import{ArrayStorage as D}from"../common/storage.js";import{NDArrayCore as h}from"../common/ndarray-core.js";import{Complex as I,isComplexLike as U}from"../common/complex.js";import{getTypedArrayConstructor as w,isBigIntDType as C,isComplexDType as S,DEFAULT_DTYPE as m}from"../common/dtype.js";function B(r){return new h(r)}function z(r){return!!(r instanceof h||r&&typeof r=="object"&&"storage"in r&&r.storage instanceof D)}function N(r,e=m){const t=D.zeros(r,e);return new h(t)}function v(r,e=m){const t=D.ones(r,e);return new h(t)}function k(r,e=m){return N(r,e)}function F(r,e,t){let n=t;n||(typeof e=="bigint"?n="int64":typeof e=="boolean"?n="bool":Number.isInteger(e)?n="int32":n=m);const o=w(n);if(!o)throw new Error(`Cannot create full array with dtype ${n}`);const u=r.reduce((s,c)=>s*c,1),a=new o(u);if(C(n)){const s=typeof e=="bigint"?e:BigInt(Math.round(Number(e)));a.fill(s)}else n==="bool"?a.fill(e?1:0):a.fill(Number(e));const f=D.fromData(a,r,n);return new h(f)}function $(r){const e=[];let t=r;for(;Array.isArray(t);)e.push(t.length),t=t[0];return e}function M(r){return typeof r=="bigint"?!0:Array.isArray(r)?r.some(e=>M(e)):!1}function E(r){return U(r)?!0:Array.isArray(r)?r.some(e=>E(e)):!1}function j(r){const e=[];function t(n){Array.isArray(n)?n.forEach(o=>t(o)):e.push(n)}return t(r),e}function b(r,e){if(r instanceof h)return!e||r.dtype===e?r.copy():r.astype(e);const t=M(r),n=E(r),o=$(r),u=o.reduce((A,p)=>A*p,1);let a=e;a||(n?a="complex128":t?a="int64":a=m);const f=S(a),s=w(a);if(!s)throw new Error(`Cannot create array with dtype ${a}`);const c=f?u*2:u,i=new s(c),y=j(r);if(C(a)){const A=i;for(let p=0;p<u;p++){const g=y[p];A[p]=typeof g=="bigint"?g:BigInt(Math.round(Number(g)))}}else if(a==="bool"){const A=i;for(let p=0;p<u;p++)A[p]=y[p]?1:0}else if(f){const A=i;for(let p=0;p<u;p++){const g=y[p];let d,x;g instanceof I?(d=g.re,x=g.im):typeof g=="object"&&g!==null&&"re"in g?(d=g.re,x=g.im??0):(d=Number(g),x=0),A[p*2]=d,A[p*2+1]=x}}else{const A=i;for(let p=0;p<u;p++){const g=y[p];A[p]=Number(g)}}const l=D.fromData(i,o,a);return new h(l)}function Y(r,e,t=1,n=m){let o=r,u=e;if(e===void 0&&(o=0,u=r),u===void 0)throw new Error("stop is required");const a=Math.max(0,Math.ceil((u-o)/t)),f=w(n);if(!f)throw new Error(`Cannot create arange array with dtype ${n}`);const s=new f(a);if(C(n))for(let i=0;i<a;i++)s[i]=BigInt(Math.round(o+i*t));else if(n==="bool")for(let i=0;i<a;i++)s[i]=o+i*t!==0?1:0;else for(let i=0;i<a;i++)s[i]=o+i*t;const c=D.fromData(s,[a],n);return new h(c)}function K(r,e,t=50,n=m){if(t<0)throw new Error("num must be non-negative");if(t===0)return b([],n);if(t===1)return b([r],n);const o=w(n);if(!o)throw new Error(`Cannot create linspace array with dtype ${n}`);const u=new o(t),a=(e-r)/(t-1);if(C(n))for(let s=0;s<t;s++)u[s]=BigInt(Math.round(r+s*a));else if(n==="bool")for(let s=0;s<t;s++)u[s]=r+s*a!==0?1:0;else for(let s=0;s<t;s++)u[s]=r+s*a;const f=D.fromData(u,[t],n);return new h(f)}function R(r,e,t=50,n=10,o=m){if(t<0)throw new Error("num must be non-negative");if(t===0)return b([],o);if(t===1)return b([Math.pow(n,r)],o);const u=w(o);if(!u)throw new Error(`Cannot create logspace array with dtype ${o}`);const a=new u(t),f=(e-r)/(t-1);if(C(o))for(let c=0;c<t;c++){const i=r+c*f;a[c]=BigInt(Math.round(Math.pow(n,i)))}else if(o==="bool")for(let c=0;c<t;c++){const i=r+c*f;a[c]=Math.pow(n,i)!==0?1:0}else for(let c=0;c<t;c++){const i=r+c*f;a[c]=Math.pow(n,i)}const s=D.fromData(a,[t],o);return new h(s)}function H(r,e,t=50,n=m){if(t<0)throw new Error("num must be non-negative");if(r===0||e===0)throw new Error("Geometric sequence cannot include zero");if(t===0)return b([],n);if(t===1)return b([r],n);const o=Math.sign(r),u=Math.sign(e);if(o!==u)throw new Error("Geometric sequence cannot contain both positive and negative values");const a=w(n);if(!a)throw new Error(`Cannot create geomspace array with dtype ${n}`);const f=new a(t),s=Math.log(Math.abs(r)),i=(Math.log(Math.abs(e))-s)/(t-1);if(C(n))for(let l=0;l<t;l++){const A=o*Math.exp(s+l*i);f[l]=BigInt(Math.round(A))}else if(n==="bool")for(let l=0;l<t;l++){const A=o*Math.exp(s+l*i);f[l]=A!==0?1:0}else for(let l=0;l<t;l++){const A=o*Math.exp(s+l*i);f[l]=A}const y=D.fromData(f,[t],n);return new h(y)}function L(r,e,t=0,n=m){const o=e??r,u=N([r,o],n),a=u.data;if(C(n)){const f=a;for(let s=0;s<r;s++){const c=s+t;c>=0&&c<o&&(f[s*o+c]=BigInt(1))}}else{const f=a;for(let s=0;s<r;s++){const c=s+t;c>=0&&c<o&&(f[s*o+c]=1)}}return u}function J(r,e=m){return L(r,r,0,e)}function T(r,e){return z(r)?!e||r.dtype===e?r:r.astype(e):b(r,e)}function Q(r,e){return N(Array.from(r.shape),e??r.dtype)}function W(r,e){return v(Array.from(r.shape),e??r.dtype)}function X(r,e){return k(Array.from(r.shape),e??r.dtype)}function Z(r,e,t){return F(Array.from(r.shape),e,t??r.dtype)}function rr(r){return r.copy()}function er(r,e){return T(r,e)}function tr(r,e){return T(r,e).copy()}function nr(r,e){return T(r,e).copy()}function or(r,e){const t=T(r,e),n=t.data;if(n instanceof Float64Array){const o=new Uint32Array(n.buffer,n.byteOffset,n.length*2),u=2146435072;for(let a=1;a<o.length;a+=2)if((o[a]&u)===u)throw new Error("array must not contain infs or NaNs")}else if(n instanceof Float32Array){const o=new Uint32Array(n.buffer,n.byteOffset,n.length),u=2139095040;for(let a=0;a<o.length;a++)if((o[a]&u)===u)throw new Error("array must not contain infs or NaNs")}else if(t.dtype==="float16")for(let o=0;o<n.length;o++){const u=Number(n[o]);if(!isFinite(u))throw new Error("array must not contain infs or NaNs")}return t}function ar(r,e,t){let n=r;return e&&e!==r.dtype&&(n=n.astype(e)),n}function q(r){const e=r.data,t=D.fromData(e.slice(),[e.length],r.dtype);return B(t)}function _(r,e=0){const t=r.shape,n=r.data,o=r.dtype;if(t.length===1){const u=t[0],a=u+Math.abs(e),f=N([a,a],o),s=f.data;for(let c=0;c<u;c++){const i=e>=0?c:c-e,y=e>=0?c+e:c;i>=0&&i<a&&y>=0&&y<a&&(s[i*a+y]=n[c])}return f}else if(t.length===2){const u=t[0],a=t[1],f=Math.min(e>=0?Math.min(u,a-e):Math.min(u+e,a),Math.max(0,e>=0?a-e:u+e));if(f<=0)return b([],o);const s=[];for(let c=0;c<f;c++){const i=e>=0?c:c-e,y=e>=0?c+e:c;i>=0&&i<u&&y>=0&&y<a&&s.push(n[i*a+y])}return b(s,o)}throw new Error("Input must be 1-D or 2-D")}function sr(r,e=0){const t=q(r);return _(t,e)}function ir(r,e,t=0,n=m){const o=e??r,u=N([r,o],n),a=u.data;for(let f=0;f<r;f++)for(let s=0;s<=Math.min(f+t,o-1);s++)s>=0&&(a[f*o+s]=1);return u}function cr(r,e=0){const t=r.shape;if(t.length<2)throw new Error("Input must be at least 2-D");const n=r.copy(),o=n.data,u=t[t.length-2],a=t[t.length-1],f=t.slice(0,-2).reduce((i,y)=>i*y,1),s=u*a,c=o instanceof BigInt64Array||o instanceof BigUint64Array;for(let i=0;i<f;i++){const y=i*s;for(let l=0;l<u;l++){const A=y+l*a+Math.max(0,Math.min(l+e+1,a)),p=y+l*a+a;A<p&&o.fill(c?0n:0,A,p)}}return n}function ur(r,e=0){const t=r.shape;if(t.length<2)throw new Error("Input must be at least 2-D");const n=r.copy(),o=n.data,u=t[t.length-2],a=t[t.length-1],f=t.slice(0,-2).reduce((i,y)=>i*y,1),s=u*a,c=o instanceof BigInt64Array||o instanceof BigUint64Array;for(let i=0;i<f;i++){const y=i*s;for(let l=0;l<u;l++){const A=y+l*a+Math.max(0,Math.min(l+e,a)),p=y+l*a;p<A&&o.fill(c?0n:0,p,A)}}return n}function fr(r,e,t=!1){const n=r.size,o=e??n,u=r.data,a=N([n,o],r.dtype),f=a.data;for(let s=0;s<n;s++){const c=u[s];for(let i=0;i<o;i++){const y=t?i:o-1-i;f[s*o+i]=Math.pow(c,y)}}return a}function lr(r,e=m,t=-1,n=0){const o=w(e);if(!o)throw new Error(`Unsupported dtype: ${e}`);let u;if(r instanceof ArrayBuffer){const f=n,s=t<0?(r.byteLength-f)/o.BYTES_PER_ELEMENT:t;u=new o(r,f,s)}else{const f=n,s=t<0?r.length:n+t,c=Array.from(r.slice(f,s));u=new o(c.length);for(let i=0;i<c.length;i++)u[i]=c[i]}const a=D.fromData(u,[u.length],e);return B(a)}function yr(r,e,t=m){const n=e.reduce((c,i)=>c*i,1),o=w(t);if(!o)throw new Error(`Unsupported dtype: ${t}`);const u=new o(n),a=[];let f=1;for(let c=e.length-1;c>=0;c--)a.unshift(f),f*=e[c];for(let c=0;c<n;c++){const i=[];let y=c;for(let l=0;l<e.length;l++)i.push(Math.floor(y/a[l])),y=y%a[l];u[c]=r(...i)}const s=D.fromData(u,e,t);return B(s)}function pr(r,e=m,t=-1){const n=[];let o=0;for(const u of r){if(t>=0&&o>=t)break;n.push(u),o++}return b(n,e)}function Ar(r,e=m,t=-1,n){const o=n??/\s+/,a=r.split(o).filter(s=>s.trim()!=="").map(s=>parseFloat(s.trim())),f=t>=0?a.slice(0,t):a;return b(f,e)}function gr(r,e=m,t=-1,n=""){throw new Error("fromfile requires Node.js file system access")}function mr(...r){if(r.length===0)return[];if(r.length===1)return[r[0].copy()];const e=r.map(o=>o.size),t=[...e],n=[];for(let o=0;o<r.length;o++){const u=r[o],a=u.data,f=N(t,u.dtype),s=f.data,c=[];let i=1;for(let l=t.length-1;l>=0;l--)c.unshift(i),i*=t[l];const y=t.reduce((l,A)=>l*A,1);for(let l=0;l<y;l++){const A=Math.floor(l/c[o])%e[o];s[l]=a[A]}n.push(f)}return n}export{Y as arange,b as array,er as asanyarray,T as asarray,or as asarray_chkfinite,tr as ascontiguousarray,nr as asfortranarray,rr as copy,_ as diag,sr as diagflat,k as empty,X as empty_like,L as eye,lr as frombuffer,gr as fromfile,yr as fromfunction,pr as fromiter,Ar as fromstring,F as full,Z as full_like,H as geomspace,J as identity,K as linspace,R as logspace,mr as meshgrid,v as ones,W as ones_like,ar as require,ir as tri,cr as tril,ur as triu,fr as vander,N as zeros,Q as zeros_like};
+import{ArrayStorage as h}from"../common/storage.js";import{NDArrayCore as D}from"../common/ndarray-core.js";import{Complex as M,isComplexLike as E}from"../common/complex.js";import{wasmAllFinite as U}from"../common/wasm/all_finite.js";import{getTypedArrayConstructor as S,isBigIntDType as N,isComplexDType as C,DEFAULT_DTYPE as m}from"../common/dtype.js";function T(r){return new D(r)}function k(r){return!!(r instanceof D||r&&typeof r=="object"&&"storage"in r&&r.storage instanceof h)}function d(r,t=m){const e=h.zeros(r,t);return new D(e)}function v(r,t=m){const e=h.ones(r,t);return new D(e)}function z(r,t=m){return new D(h.empty(r,t))}function L(r,t,e){let a=e;a||(typeof t=="bigint"?a="int64":typeof t=="boolean"?a="bool":Number.isInteger(t)?a=t>=-2147483648&&t<=2147483647?"int32":"int64":a=m);const o=h.empty(r,a),c=o.data;if(N(a)){const n=typeof t=="bigint"?t:BigInt(Math.round(Number(t)));c.fill(n)}else if(a==="bool")c.fill(t?1:0);else if(C(a)){const n=Number(t);for(let i=0;i<c.length;i+=2)c[i]=n,c[i+1]=0}else c.fill(Number(t));return new D(o)}function j(r){return Array.isArray(r)||ArrayBuffer.isView(r)}function q(r){const t=[];let e=r;for(;j(e);)t.push(e.length),e=e[0];return t}function F(r){return typeof r=="bigint"||r instanceof BigInt64Array||r instanceof BigUint64Array?!0:Array.isArray(r)?r.some(t=>F(t)):!1}function I(r){return E(r)?!0:Array.isArray(r)?r.some(t=>I(t)):!1}function V(r){const t=[];function e(a){if(Array.isArray(a))a.forEach(o=>e(o));else if(ArrayBuffer.isView(a)&&"length"in a){const o=a;for(let c=0;c<o.length;c++)t.push(o[c])}else t.push(a)}return e(r),t}function x(r,t){if(r instanceof D)return!t||r.dtype===t?r.copy():r.astype(t);const e=F(r),a=I(r),o=q(r),c=o.reduce((f,l)=>f*l,1);let n=t;n||(r instanceof Float64Array?n="float64":r instanceof Float32Array?n="float32":r instanceof Int32Array?n="int32":r instanceof Int16Array?n="int16":r instanceof Int8Array?n="int8":r instanceof Uint32Array?n="uint32":r instanceof Uint16Array?n="uint16":r instanceof Uint8Array||r instanceof Uint8ClampedArray?n="uint8":r instanceof BigInt64Array?n="int64":r instanceof BigUint64Array?n="uint64":typeof Float16Array<"u"&&r instanceof Float16Array?n="float16":a?n="complex128":e?n="int64":n=m);const i=C(n),y=h.empty(o,n),u=y.data,s=V(r);if(N(n)){const f=u;for(let l=0;l<c;l++){const p=s[l];f[l]=typeof p=="bigint"?p:BigInt(Math.trunc(Number(p)))}}else if(n==="bool"){const f=u;for(let l=0;l<c;l++)f[l]=s[l]?1:0}else if(i){const f=u;for(let l=0;l<c;l++){const p=s[l];let A,g;p instanceof M?(A=p.re,g=p.im):typeof p=="object"&&p!==null&&"re"in p?(A=p.re,g=p.im??0):(A=Number(p),g=0),f[l*2]=A,f[l*2+1]=g}}else{const f=u;for(let l=0;l<c;l++){const p=s[l];f[l]=Number(p)}}return new D(y)}function H(r,t,e=1,a){let o=r,c=t;if(t===void 0&&(o=0,c=r),c===void 0)throw new Error("stop is required");const n=Math.max(0,Math.ceil((c-o)/e)),i=a??m;if(i==="bool"&&n>2)throw new TypeError("arange() is only supported for booleans when the result has at most length 2.");const y=h.empty([n],i),u=y.data;if(N(i))for(let s=0;s<n;s++)u[s]=BigInt(Math.trunc(o+s*e));else if(i==="bool")for(let s=0;s<n;s++)u[s]=o+s*e!==0?1:0;else if(C(i))for(let s=0;s<n;s++)u[s*2]=o+s*e,u[s*2+1]=0;else for(let s=0;s<n;s++)u[s]=o+s*e;return new D(y)}function J(r,t,e=50,a=m){if(e<0)throw new Error("num must be non-negative");if(e===0)return x([],a);if(e===1)return x([r],a);const o=h.empty([e],a),c=o.data,n=(t-r)/(e-1);if(N(a))for(let i=0;i<e;i++)c[i]=BigInt(Math.trunc(r+i*n));else if(a==="bool")for(let i=0;i<e;i++)c[i]=r+i*n!==0?1:0;else if(C(a))for(let i=0;i<e;i++)c[i*2]=r+i*n,c[i*2+1]=0;else for(let i=0;i<e;i++)c[i]=r+i*n;return new D(o)}function Q(r,t,e=50,a=10,o=m){if(e<0)throw new Error("num must be non-negative");if(e===0)return x([],o);if(e===1)return x([Math.pow(a,r)],o);const c=h.empty([e],o),n=c.data,i=(t-r)/(e-1);if(N(o))for(let y=0;y<e;y++){const u=r+y*i;n[y]=BigInt(Math.trunc(Math.pow(a,u)))}else if(o==="bool")for(let y=0;y<e;y++){const u=r+y*i;n[y]=Math.pow(a,u)!==0?1:0}else if(C(o))for(let y=0;y<e;y++){const u=r+y*i;n[y*2]=Math.pow(a,u),n[y*2+1]=0}else for(let y=0;y<e;y++){const u=r+y*i;n[y]=Math.pow(a,u)}return new D(c)}function W(r,t,e=50,a=m){if(e<0)throw new Error("num must be non-negative");if(r===0||t===0)throw new Error("Geometric sequence cannot include zero");if(e===0)return x([],a);if(e===1)return x([r],a);const o=Math.sign(r),c=Math.sign(t);if(o!==c)throw new Error("Geometric sequence cannot contain both positive and negative values");const n=h.empty([e],a),i=n.data,y=Math.log(Math.abs(r)),s=(Math.log(Math.abs(t))-y)/(e-1);if(N(a))for(let f=0;f<e;f++){const l=o*Math.exp(y+f*s);i[f]=BigInt(Math.trunc(l))}else if(a==="bool")for(let f=0;f<e;f++){const l=o*Math.exp(y+f*s);i[f]=l!==0?1:0}else if(C(a))for(let f=0;f<e;f++){const l=o*Math.exp(y+f*s);i[f*2]=l,i[f*2+1]=0}else for(let f=0;f<e;f++){const l=o*Math.exp(y+f*s);i[f]=l}return new D(n)}function O(r,t,e=0,a=m){const o=t??r,c=d([r,o],a),n=c.data;if(N(a)){const i=n;for(let y=0;y<r;y++){const u=y+e;u>=0&&u<o&&(i[y*o+u]=BigInt(1))}}else{const i=n;for(let y=0;y<r;y++){const u=y+e;u>=0&&u<o&&(i[y*o+u]=1)}}return c}function X(r,t=m){return O(r,r,0,t)}function B(r,t){return k(r)?!t||r.dtype===t?r:r.astype(t):x(r,t)}function Z(r,t){return d(Array.from(r.shape),t??r.dtype)}function rr(r,t){return v(Array.from(r.shape),t??r.dtype)}function tr(r,t){return z(Array.from(r.shape),t??r.dtype)}function er(r,t,e){return L(Array.from(r.shape),t,e??r.dtype)}function nr(r){return r.copy()}function or(r,t){return B(r,t)}function ar(r,t){return B(r,t).copy()}function sr(r,t){return B(r,t).copy()}function ir(r,t){const e=B(r,t),a=U(e.storage);if(a===!0)return e;if(a===!1)throw new Error("array must not contain infs or NaNs");const o=e.data;if(o instanceof Float64Array){const c=new Uint32Array(o.buffer,o.byteOffset,o.length*2),n=2146435072;for(let i=1;i<c.length;i+=2)if((c[i]&n)===n)throw new Error("array must not contain infs or NaNs")}else if(o instanceof Float32Array){const c=new Uint32Array(o.buffer,o.byteOffset,o.length),n=2139095040;for(let i=0;i<c.length;i++)if((c[i]&n)===n)throw new Error("array must not contain infs or NaNs")}else if(e.dtype==="float16"){const c=new Uint16Array(o.buffer,o.byteOffset,o.length),n=31744;for(let i=0;i<c.length;i++)if((c[i]&n)===n)throw new Error("array must not contain infs or NaNs")}return e}function lr(r,t,e){let a=r;return t&&t!==r.dtype&&(a=a.astype(t)),a}function _(r){const t=r.data,e=r.size,a=h.empty([e],r.dtype);return a.data.set(t),T(a)}function G(r,t=0){const e=r.shape,a=r.data,o=r.dtype;if(e.length===1){const c=e[0],n=c+Math.abs(t),i=d([n,n],o),y=i.data,u=C(o);for(let s=0;s<c;s++){const f=t>=0?s:s-t,l=t>=0?s+t:s;if(f>=0&&f<n&&l>=0&&l<n)if(u){const p=s*2,A=(f*n+l)*2;y[A]=a[p],y[A+1]=a[p+1]}else y[f*n+l]=a[s]}return i}else if(e.length===2){const c=e[0],n=e[1],i=Math.min(t>=0?Math.min(c,n-t):Math.min(c+t,n),Math.max(0,t>=0?n-t:c+t));if(i<=0)return x([],o);if(C(o)){const s=h.empty([i],o),f=s.data;for(let l=0;l<i;l++){const p=t>=0?l:l-t,A=t>=0?l+t:l;if(p>=0&&p<c&&A>=0&&A<n){const g=(p*n+A)*2;f[l*2]=a[g],f[l*2+1]=a[g+1]}}return new D(s)}const u=[];for(let s=0;s<i;s++){const f=t>=0?s:s-t,l=t>=0?s+t:s;f>=0&&f<c&&l>=0&&l<n&&u.push(a[f*n+l])}return x(u,o)}throw new Error("Input must be 1-D or 2-D")}function cr(r,t=0){const e=_(r);return G(e,t)}function fr(r,t,e=0,a=m){const o=t??r,c=d([r,o],a),n=c.data,i=a==="complex128"||a==="complex64",y=a==="int64"||a==="uint64";if(i){const u=n;for(let s=0;s<r;s++)for(let f=0;f<=Math.min(s+e,o-1);f++)if(f>=0){const l=(s*o+f)*2;u[l]=1,u[l+1]=0}}else if(y){const u=n;for(let s=0;s<r;s++)for(let f=0;f<=Math.min(s+e,o-1);f++)f>=0&&(u[s*o+f]=1n)}else for(let u=0;u<r;u++)for(let s=0;s<=Math.min(u+e,o-1);s++)s>=0&&(n[u*o+s]=1);return c}function yr(r,t=0){const e=r.shape;if(e.length<2)throw new Error("Input must be at least 2-D");const a=r.copy(),o=a.data,c=e[e.length-2],n=e[e.length-1],i=e.slice(0,-2).reduce((f,l)=>f*l,1),y=c*n,u=o instanceof BigInt64Array||o instanceof BigUint64Array,s=r.dtype==="complex128"||r.dtype==="complex64";for(let f=0;f<i;f++){const l=f*y;for(let p=0;p<c;p++){const A=Math.max(0,Math.min(p+t+1,n)),g=n;if(A<g)if(s){const w=(l+p*n+A)*2,b=(l+p*n+g)*2;o.fill(0,w,b)}else{const w=l+p*n+A,b=l+p*n+g;o.fill(u?0n:0,w,b)}}}return a}function ur(r,t=0){const e=r.shape;if(e.length<2)throw new Error("Input must be at least 2-D");const a=r.copy(),o=a.data,c=e[e.length-2],n=e[e.length-1],i=e.slice(0,-2).reduce((f,l)=>f*l,1),y=c*n,u=o instanceof BigInt64Array||o instanceof BigUint64Array,s=r.dtype==="complex128"||r.dtype==="complex64";for(let f=0;f<i;f++){const l=f*y;for(let p=0;p<c;p++){const A=Math.max(0,Math.min(p+t,n)),g=0;if(g<A)if(s){const w=(l+p*n+g)*2,b=(l+p*n+A)*2;o.fill(0,w,b)}else{const w=l+p*n+g,b=l+p*n+A;o.fill(u?0n:0,w,b)}}}return a}function pr(r,t,e=!1){const a=r.size,o=t??a,c=r.data,n=r.dtype,i=n==="complex128"||n==="complex64"?"complex128":n==="float64"||n==="float32"||n==="float16"||n==="uint64"?"float64":"int64",y=d([a,o],i),u=y.data,s=i==="int64",f=i==="complex128";for(let l=0;l<a;l++){const p=Number(typeof c[l]=="bigint"?c[l]:c[f?l*2:l]);for(let A=0;A<o;A++){const g=e?A:o-1-A,w=Math.pow(p,g),b=l*o+A;s?u[b]=BigInt(Math.round(w)):f?(u[b*2]=w,u[b*2+1]=0):u[b]=w}}return y}function Ar(r,t=m,e=-1,a=0){const o=S(t);if(!o)throw new Error(`Unsupported dtype: ${t}`);let c;if(r instanceof ArrayBuffer){const i=a,y=e<0?(r.byteLength-i)/o.BYTES_PER_ELEMENT:e;c=new o(r,i,y)}else{const i=a,y=e<0?r.length:a+e,u=Array.from(r.slice(i,y));c=new o(u.length);for(let s=0;s<u.length;s++)c[s]=u[s]}const n=h.fromData(c,[c.length],t);return T(n)}function gr(r,t,e=m){const a=t.reduce((s,f)=>s*f,1);if(e==="bool"&&a>2)throw new TypeError("arange() is only supported for booleans when the result has at most length 2.");const o=h.empty(t,e),c=o.data,n=[];let i=1;for(let s=t.length-1;s>=0;s--)n.unshift(i),i*=t[s];const y=C(e),u=N(e);for(let s=0;s<a;s++){const f=[];let l=s;for(let A=0;A<t.length;A++)f.push(Math.floor(l/n[A])),l=l%n[A];const p=r(...f);y?(c[s*2]=p,c[s*2+1]=0):u?c[s]=BigInt(Math.round(p)):c[s]=p}return T(o)}function mr(r,t=m,e=-1){const a=[];let o=0;for(const c of r){if(e>=0&&o>=e)break;a.push(c),o++}return x(a,t)}function hr(r,t=m,e=-1,a){const o=a??/\s+/,n=r.split(o).filter(y=>y.trim()!=="").map(y=>parseFloat(y.trim())),i=e>=0?n.slice(0,e):n;return x(i,t)}function Dr(r,t=m,e=-1,a=""){throw new Error("fromfile requires Node.js file system access")}function br(...r){if(r.length===0)return[];if(r.length===1)return[r[0].copy()];const t=r.map(o=>o.size),e=[...t],a=[];for(let o=0;o<r.length;o++){const c=r[o],n=c.data,i=d(e,c.dtype),y=i.data,u=[];let s=1;for(let l=e.length-1;l>=0;l--)u.unshift(s),s*=e[l];const f=e.reduce((l,p)=>l*p,1);for(let l=0;l<f;l++){const p=Math.floor(l/u[o])%t[o];y[l]=n[p]}a.push(i)}return a}export{H as arange,x as array,or as asanyarray,B as asarray,ir as asarray_chkfinite,ar as ascontiguousarray,sr as asfortranarray,nr as copy,G as diag,cr as diagflat,z as empty,tr as empty_like,O as eye,Ar as frombuffer,Dr as fromfile,gr as fromfunction,mr as fromiter,hr as fromstring,L as full,er as full_like,W as geomspace,X as identity,J as linspace,Q as logspace,br as meshgrid,v as ones,rr as ones_like,lr as require,fr as tri,yr as tril,ur as triu,pr as vander,d as zeros,Z as zeros_like};

package/dist/esm/core/index.js

@@ -1 +1 @@
-import{NDArrayCore as w,Complex as A,ArrayStorage as v}from"./types.js";import{zeros as k,ones as z,empty as q,full as N,array as I,arange as D,linspace as T,logspace as C,geomspace as O,eye as P,identity as E,asarray as L,zeros_like as j,ones_like as R,empty_like as G,full_like as H,copy as U,asanyarray as Y,ascontiguousarray as M,asfortranarray as V,asarray_chkfinite as B,require as F,diag as J,diagflat as K,tri as Q,tril as W,triu as X,vander as Z,frombuffer as $,fromfunction as rr,fromiter as er,fromstring as nr,fromfile as ar,meshgrid as tr}from"./creation.js";import{add as mr,subtract as ir,multiply as ur,sqrt as sr,power as br,pow as fr,exp as pr,exp2 as lr,expm1 as dr,log as cr,log2 as _r,log10 as gr,log1p as yr,logaddexp as hr,logaddexp2 as xr,absolute as wr,abs as Ar,negative as vr,sign as Sr,mod as kr,divide as zr,true_divide as qr,floor_divide as Nr,positive as Ir,reciprocal as Dr,cbrt as Tr,fabs as Cr,divmod as Or,square as Pr,remainder as Er,heaviside as Lr,float_power as jr,fmod as Rr,frexp as Gr,gcd as Hr,lcm as Ur,ldexp as Yr,modf as Mr,clip as Vr,maximum as Br,minimum as Fr,fmax as Jr,fmin as Kr,nan_to_num as Qr,interp as Wr,unwrap as Xr,sinc as Zr,i0 as $r}from"./arithmetic.js";import{sin as ee,cos as ne,tan as ae,arcsin as te,asin as oe,arccos as me,acos as ie,arctan as ue,atan as se,arctan2 as be,atan2 as fe,hypot as pe,degrees as le,radians as de,deg2rad as ce,rad2deg as _e,sinh as ge,cosh as ye,tanh as he,arcsinh as xe,asinh as we,arccosh as Ae,acosh as ve,arctanh as Se,atanh as ke}from"./trig.js";import{dot as qe,matmul as Ne,trace as Ie,diagonal as De,kron as Te,transpose as Ce,inner as Oe,outer as Pe,tensordot as Ee,einsum as Le,einsum_path as je,vdot as Re,vecdot as Ge,matrix_transpose as He,permute_dims as Ue,matvec as Ye,vecmat as Me,cross as Ve,linalg as Be}from"./linalg.js";import{reshape as Je,flatten as Ke,ravel as Qe,squeeze as We,expand_dims as Xe,swapaxes as Ze,moveaxis as $e,rollaxis as rn,concatenate as en,stack as nn,vstack as an,hstack as tn,dstack as on,concat as mn,column_stack as un,row_stack as sn,block as bn,split as fn,array_split as pn,vsplit as ln,hsplit as dn,dsplit as cn,unstack as _n,tile as gn,repeat as yn,flip as hn,fliplr as xn,flipud as wn,rot90 as An,roll as vn,resize as Sn,atleast_1d as kn,atleast_2d as zn,atleast_3d as qn}from"./shape.js";import{sum as In,mean as Dn,prod as Tn,max as Cn,amax as On,min as Pn,amin as En,ptp as Ln,argmin as jn,argmax as Rn,variance as Gn,var_ as Hn,variance as Un,std as Yn,median as Mn,percentile as Vn,quantile as Bn,average as Fn,all as Jn,any as Kn,cumsum as Qn,cumulative_sum as Wn,cumprod as Xn,cumulative_prod as Zn,nansum as $n,nanprod as ra,nanmean as ea,nanvar as na,nanstd as aa,nanmin as ta,nanmax as oa,nanargmin as ma,nanargmax as ia,nancumsum as ua,nancumprod as sa,nanmedian as ba,nanquantile as fa,nanpercentile as pa}from"./reduction.js";import{logical_and as da,logical_or as ca,logical_not as _a,logical_xor as ga,isfinite as ya,isinf as ha,isnan as xa,isnat as wa,isneginf as Aa,isposinf as va,iscomplex as Sa,iscomplexobj as ka,isreal as za,isrealobj as qa,real_if_close as Na,isfortran as Ia,isscalar as Da,iterable as Ta,isdtype as Ca,promote_types as Oa,copysign as Pa,signbit as Ea,nextafter as La,spacing as ja,greater as Ra,greater_equal as Ga,less as Ha,less_equal as Ua,equal as Ya,not_equal as Ma,isclose as Va,allclose as Ba}from"./logic.js";import{sort as Ja,argsort as Ka,lexsort as Qa,partition as Wa,argpartition as Xa,sort_complex as Za,nonzero as $a,argwhere as rt,flatnonzero as et,where as nt,searchsorted as at,extract as tt,count_nonzero as ot}from"./sorting.js";import{bitwise_and as it,bitwise_or as ut,bitwise_xor as st,bitwise_not as bt,invert as ft,left_shift as pt,right_shift as lt,packbits as dt,unpackbits as ct,bitwise_count as _t,bitwise_invert as gt,bitwise_left_shift as yt,bitwise_right_shift as ht}from"./bitwise.js";import{around as wt,round as At,ceil as vt,fix as St,floor as kt,rint as zt,trunc as qt}from"./rounding.js";import{unique as It,in1d as Dt,intersect1d as Tt,isin as Ct,setdiff1d as Ot,setxor1d as Pt,union1d as Et,trim_zeros as Lt,unique_all as jt,unique_counts as Rt,unique_inverse as Gt,unique_values as Ht}from"./sets.js";import{bincount as Yt,digitize as Mt,histogram as Vt,histogram2d as Bt,histogramdd as Ft,correlate as Jt,convolve as Kt,cov as Qt,corrcoef as Wt,histogram_bin_edges as Xt,trapezoid as Zt}from"./statistics.js";import{diff as ro,ediff1d as eo,gradient as no}from"./gradient.js";import{real as to,imag as oo,conj as mo,conjugate as io,angle as uo}from"./complex.js";import{broadcast_to as bo,broadcast_arrays as fo,broadcast_shapes as po,take as lo,put as co,take_along_axis as _o,put_along_axis as go,choose as yo,compress as ho,select as xo,place as wo,putmask as Ao,copyto as vo,indices as So,ix_ as ko,ravel_multi_index as zo,unravel_index as qo,diag_indices as No,diag_indices_from as Io,fill_diagonal as Do,tril_indices as To,tril_indices_from as Co,triu_indices as Oo,triu_indices_from as Po,mask_indices as Eo,array_equal as Lo,array_equiv as jo,apply_along_axis as Ro,apply_over_axes as Go,may_share_memory as Ho,shares_memory as Uo,geterr as Yo,seterr as Mo,iindex as Vo,bindex as Bo}from"./advanced.js";import{set_printoptions as Jo,get_printoptions as Ko,printoptions as Qo,format_float_positional as Wo,format_float_scientific as Xo,base_repr as Zo,binary_repr as $o,array2string as rm,array_repr as em,array_str as nm}from"./formatting.js";import{ndim as tm,shape as om,size as mm,item as im,tolist as um,tobytes as sm,byteswap as bm,view as fm,tofile as pm,fill as lm}from"./utility.js";import{append as cm,delete_ as _m,delete_ as gm,insert as ym,pad as hm}from"./shape-extra.js";import{can_cast as wm,common_type as Am,result_type as vm,min_scalar_type as Sm,issubdtype as km,typename as zm,mintypecode as qm}from"./typechecking.js";import{poly as Im,polyadd as Dm,polyder as Tm,polydiv as Cm,polyfit as Om,polyint as Pm,polymul as Em,polysub as Lm,polyval as jm,roots as Rm}from"./polynomial.js";import{parseNpy as Hm,parseNpyHeader as Um,parseNpyData as Ym}from"../io/npy/parser.js";import{serializeNpy as Vm}from"../io/npy/serializer.js";import{UnsupportedDTypeError as Fm,InvalidNpyError as Jm,SUPPORTED_DTYPES as Km,DTYPE_TO_DESCR as Qm}from"../io/npy/format.js";import{parseNpz as Xm,parseNpzSync as Zm,loadNpz as $m,loadNpzSync as ri}from"../io/npz/parser.js";import{serializeNpz as ni,serializeNpzSync as ai}from"../io/npz/serializer.js";import{parseTxt as oi,genfromtxt as mi,fromregex as ii}from"../io/txt/parser.js";import{serializeTxt as si}from"../io/txt/serializer.js";import*as o from"../common/ops/random.js";import{NDArrayCore as m}from"../common/ndarray-core.js";function t(r){return r&&typeof r=="object"&&"_data"in r&&"_shape"in r?m.fromStorage(r):r}const g={seed:o.seed,get_state:o.get_state,set_state:o.set_state,get_bit_generator:o.get_bit_generator,set_bit_generator:o.set_bit_generator,default_rng:r=>{const e=o.default_rng(r),n=a=>a instanceof m?a.storage:a;return{random:a=>t(e.random(a)),integers:(a,s,b)=>t(e.integers(a,s,b)),standard_normal:a=>t(e.standard_normal(a)),normal:(a,s,b)=>t(e.normal(a,s,b)),uniform:(a,s,b)=>t(e.uniform(a,s,b)),choice:(a,s,b,f)=>t(e.choice(n(a),s,b,f instanceof m?f.storage:f)),permutation:a=>t(e.permutation(n(a))),shuffle:a=>e.shuffle(a instanceof m?a.storage:a),exponential:(a,s)=>t(e.exponential(a,s)),poisson:(a,s)=>t(e.poisson(a,s)),binomial:(a,s,b)=>t(e.binomial(a,s,b))}},Generator:o.Generator,random:r=>t(o.random(r)),rand:(...r)=>t(o.rand(...r)),randn:(...r)=>t(o.randn(...r)),randint:(r,e,n,a)=>t(o.randint(r,e,n,a)),random_sample:r=>t(o.random_sample(r)),ranf:r=>t(o.ranf(r)),sample:r=>t(o.sample(r)),random_integers:(r,e,n)=>t(o.random_integers(r,e,n)),bytes:o.bytes,uniform:(r,e,n)=>t(o.uniform(r,e,n)),normal:(r,e,n)=>t(o.normal(r,e,n)),standard_normal:r=>t(o.standard_normal(r)),exponential:(r,e)=>t(o.exponential(r,e)),standard_exponential:r=>t(o.standard_exponential(r)),gamma:(r,e,n)=>t(o.gamma(r,e,n)),standard_gamma:(r,e)=>t(o.standard_gamma(r,e)),beta:(r,e,n)=>t(o.beta(r,e,n)),chisquare:(r,e)=>t(o.chisquare(r,e)),noncentral_chisquare:(r,e,n)=>t(o.noncentral_chisquare(r,e,n)),f:(r,e,n)=>t(o.f(r,e,n)),noncentral_f:(r,e,n,a)=>t(o.noncentral_f(r,e,n,a)),standard_cauchy:r=>t(o.standard_cauchy(r)),standard_t:(r,e)=>t(o.standard_t(r,e)),laplace:(r,e,n)=>t(o.laplace(r,e,n)),logistic:(r,e,n)=>t(o.logistic(r,e,n)),lognormal:(r,e,n)=>t(o.lognormal(r,e,n)),gumbel:(r,e,n)=>t(o.gumbel(r,e,n)),pareto:(r,e)=>t(o.pareto(r,e)),power:(r,e)=>t(o.power(r,e)),rayleigh:(r,e)=>t(o.rayleigh(r,e)),triangular:(r,e,n,a)=>t(o.triangular(r,e,n,a)),wald:(r,e,n)=>t(o.wald(r,e,n)),weibull:(r,e)=>t(o.weibull(r,e)),poisson:(r,e)=>t(o.poisson(r,e)),binomial:(r,e,n)=>t(o.binomial(r,e,n)),geometric:(r,e)=>t(o.geometric(r,e)),hypergeometric:(r,e,n,a)=>t(o.hypergeometric(r,e,n,a)),logseries:(r,e)=>t(o.logseries(r,e)),negative_binomial:(r,e,n)=>t(o.negative_binomial(r,e,n)),zipf:(r,e)=>t(o.zipf(r,e)),multinomial:(r,e,n)=>t(o.multinomial(r,e,n)),multivariate_normal:(r,e,n,a,s)=>t(o.multivariate_normal(r,e,n,a,s)),dirichlet:(r,e)=>t(o.dirichlet(r,e)),vonmises:(r,e,n)=>t(o.vonmises(r,e,n)),choice:(r,e,n,a)=>t(o.choice(r,e,n,a)),permutation:r=>t(o.permutation(r)),shuffle:r=>o.shuffle(r instanceof m?r.storage:r)};import*as i from"../common/ops/fft.js";function u(r){return r instanceof m?r.storage:r}const y={fft:(r,e,n,a)=>m.fromStorage(i.fft(u(r),e,n,a)),ifft:(r,e,n,a)=>m.fromStorage(i.ifft(u(r),e,n,a)),fft2:(r,e,n,a)=>m.fromStorage(i.fft2(u(r),e,n,a)),ifft2:(r,e,n,a)=>m.fromStorage(i.ifft2(u(r),e,n,a)),fftn:(r,e,n,a)=>m.fromStorage(i.fftn(u(r),e,n,a)),ifftn:(r,e,n,a)=>m.fromStorage(i.ifftn(u(r),e,n,a)),rfft:(r,e,n,a)=>m.fromStorage(i.rfft(u(r),e,n,a)),irfft:(r,e,n,a)=>m.fromStorage(i.irfft(u(r),e,n,a)),rfft2:(r,e,n,a)=>m.fromStorage(i.rfft2(u(r),e,n,a)),irfft2:(r,e,n,a)=>m.fromStorage(i.irfft2(u(r),e,n,a)),rfftn:(r,e,n,a)=>m.fromStorage(i.rfftn(u(r),e,n,a)),irfftn:(r,e,n,a)=>m.fromStorage(i.irfftn(u(r),e,n,a)),hfft:(r,e,n,a)=>m.fromStorage(i.hfft(u(r),e,n,a)),ihfft:(r,e,n,a)=>m.fromStorage(i.ihfft(u(r),e,n,a)),fftfreq:(r,e)=>m.fromStorage(i.fftfreq(r,e)),rfftfreq:(r,e)=>m.fromStorage(i.rfftfreq(r,e)),fftshift:(r,e)=>m.fromStorage(i.fftshift(u(r),e)),ifftshift:(r,e)=>m.fromStorage(i.ifftshift(u(r),e))};export{v as ArrayStorage,A as Complex,Qm as DTYPE_TO_DESCR,Jm as InvalidNpyError,w as NDArrayCore,Km as SUPPORTED_DTYPES,Fm as UnsupportedDTypeError,Ar as abs,wr as absolute,ie as acos,ve as acosh,mr as add,Jn as all,Ba as allclose,On as amax,En as amin,uo as angle,Kn as any,cm as append,Ro as apply_along_axis,Go as apply_over_axes,D as arange,me as arccos,Ae as arccosh,te as arcsin,xe as arcsinh,ue as arctan,be as arctan2,Se as arctanh,Rn as argmax,jn as argmin,Xa as argpartition,Ka as argsort,rt as argwhere,wt as around,I as array,rm as array2string,Lo as array_equal,jo as array_equiv,em as array_repr,pn as array_split,nm as array_str,Y as asanyarray,L as asarray,B as asarray_chkfinite,M as ascontiguousarray,V as asfortranarray,oe as asin,we as asinh,se as atan,fe as atan2,ke as atanh,kn as atleast_1d,zn as atleast_2d,qn as atleast_3d,Fn as average,Zo as base_repr,$o as binary_repr,Yt as bincount,Bo as bindex,it as bitwise_and,_t as bitwise_count,gt as bitwise_invert,yt as bitwise_left_shift,bt as bitwise_not,ut as bitwise_or,ht as bitwise_right_shift,st as bitwise_xor,bn as block,fo as broadcast_arrays,po as broadcast_shapes,bo as broadcast_to,bm as byteswap,wm as can_cast,Tr as cbrt,vt as ceil,yo as choose,Vr as clip,un as column_stack,Am as common_type,ho as compress,mn as concat,en as concatenate,mo as conj,io as conjugate,Kt as convolve,U as copy,Pa as copysign,vo as copyto,Wt as corrcoef,Jt as correlate,ne as cos,ye as cosh,ot as count_nonzero,Qt as cov,Ve as cross,Xn as cumprod,Qn as cumsum,Zn as cumulative_prod,Wn as cumulative_sum,ce as deg2rad,le as degrees,gm as delete,_m as delete_,J as diag,No as diag_indices,Io as diag_indices_from,K as diagflat,De as diagonal,ro as diff,Mt as digitize,zr as divide,Or as divmod,qe as dot,cn as dsplit,on as dstack,eo as ediff1d,Le as einsum,je as einsum_path,q as empty,G as empty_like,Ya as equal,pr as exp,lr as exp2,Xe as expand_dims,dr as expm1,tt as extract,P as eye,Cr as fabs,y as fft,lm as fill,Do as fill_diagonal,St as fix,et as flatnonzero,Ke as flatten,hn as flip,xn as fliplr,wn as flipud,jr as float_power,kt as floor,Nr as floor_divide,Jr as fmax,Kr as fmin,Rr as fmod,Wo as format_float_positional,Xo as format_float_scientific,Gr as frexp,$ as frombuffer,ar as fromfile,rr as fromfunction,er as fromiter,ii as fromregex,nr as fromstring,N as full,H as full_like,Hr as gcd,mi as genfromtxt,O as geomspace,Ko as get_printoptions,Yo as geterr,no as gradient,Ra as greater,Ga as greater_equal,Lr as heaviside,Vt as histogram,Bt as histogram2d,Xt as histogram_bin_edges,Ft as histogramdd,dn as hsplit,tn as hstack,pe as hypot,$r as i0,E as identity,Vo as iindex,oo as imag,Dt as in1d,So as indices,Oe as inner,ym as insert,Wr as interp,Tt as intersect1d,ft as invert,Va as isclose,Sa as iscomplex,ka as iscomplexobj,Ca as isdtype,ya as isfinite,Ia as isfortran,Ct as isin,ha as isinf,xa as isnan,wa as isnat,Aa as isneginf,va as isposinf,za as isreal,qa as isrealobj,Da as isscalar,km as issubdtype,im as item,Ta as iterable,ko as ix_,Te as kron,Ur as lcm,Yr as ldexp,pt as left_shift,Ha as less,Ua as less_equal,Qa as lexsort,Be as linalg,T as linspace,$m as loadNpz,ri as loadNpzSync,cr as log,gr as log10,yr as log1p,_r as log2,hr as logaddexp,xr as logaddexp2,da as logical_and,_a as logical_not,ca as logical_or,ga as logical_xor,C as logspace,Eo as mask_indices,Ne as matmul,He as matrix_transpose,Ye as matvec,Cn as max,Br as maximum,Ho as may_share_memory,Dn as mean,Mn as median,tr as meshgrid,Pn as min,Sm as min_scalar_type,Fr as minimum,qm as mintypecode,kr as mod,Mr as modf,$e as moveaxis,ur as multiply,Qr as nan_to_num,ia as nanargmax,ma as nanargmin,sa as nancumprod,ua as nancumsum,oa as nanmax,ea as nanmean,ba as nanmedian,ta as nanmin,pa as nanpercentile,ra as nanprod,fa as nanquantile,aa as nanstd,$n as nansum,na as nanvar,tm as ndim,vr as negative,La as nextafter,$a as nonzero,Ma as not_equal,z as ones,R as ones_like,Pe as outer,dt as packbits,hm as pad,Hm as parseNpy,Ym as parseNpyData,Um as parseNpyHeader,Xm as parseNpz,Zm as parseNpzSync,oi as parseTxt,Wa as partition,Vn as percentile,Ue as permute_dims,wo as place,Im as poly,Dm as polyadd,Tm as polyder,Cm as polydiv,Om as polyfit,Pm as polyint,Em as polymul,Lm as polysub,jm as polyval,Ir as positive,fr as pow,br as power,Qo as printoptions,Tn as prod,Oa as promote_types,Ln as ptp,co as put,go as put_along_axis,Ao as putmask,Bn as quantile,_e as rad2deg,de as radians,g as random,Qe as ravel,zo as ravel_multi_index,to as real,Na as real_if_close,Dr as reciprocal,Er as remainder,yn as repeat,F as require,Je as reshape,Sn as resize,vm as result_type,lt as right_shift,zt as rint,vn as roll,rn as rollaxis,Rm as roots,An as rot90,At as round,sn as row_stack,at as searchsorted,xo as select,Vm as serializeNpy,ni as serializeNpz,ai as serializeNpzSync,si as serializeTxt,Jo as set_printoptions,Ot as setdiff1d,Mo as seterr,Pt as setxor1d,om as shape,Uo as shares_memory,Sr as sign,Ea as signbit,ee as sin,Zr as sinc,ge as sinh,mm as size,Ja as sort,Za as sort_complex,ja as spacing,fn as split,sr as sqrt,Pr as square,We as squeeze,nn as stack,Yn as std,ir as subtract,In as sum,Ze as swapaxes,lo as take,_o as take_along_axis,ae as tan,he as tanh,Ee as tensordot,gn as tile,sm as tobytes,pm as tofile,um as tolist,Ie as trace,Ce as transpose,Zt as trapezoid,Q as tri,W as tril,To as tril_indices,Co as tril_indices_from,Lt as trim_zeros,X as triu,Oo as triu_indices,Po as triu_indices_from,qr as true_divide,qt as trunc,zm as typename,Et as union1d,It as unique,jt as unique_all,Rt as unique_counts,Gt as unique_inverse,Ht as unique_values,ct as unpackbits,qo as unravel_index,_n as unstack,Xr as unwrap,Z as vander,Un as var,Hn as var_,Gn as variance,Re as vdot,Ge as vecdot,Me as vecmat,fm as view,ln as vsplit,an as vstack,nt as where,k as zeros,j as zeros_like};
+import{NDArrayCore as w,Complex as A,ArrayStorage as v}from"./types.js";import{zeros as k,ones as z,empty as N,full as q,array as I,arange as D,linspace as T,logspace as C,geomspace as O,eye as P,identity as E,asarray as L,zeros_like as j,ones_like as R,empty_like as G,full_like as H,copy as U,asanyarray as Y,ascontiguousarray as M,asfortranarray as V,asarray_chkfinite as B,require as F,diag as J,diagflat as K,tri as Q,tril as W,triu as X,vander as Z,frombuffer as $,fromfunction as rr,fromiter as er,fromstring as nr,fromfile as ar,meshgrid as tr}from"./creation.js";import{add as mr,subtract as ir,multiply as ur,sqrt as sr,power as br,pow as fr,exp as pr,exp2 as lr,expm1 as dr,log as cr,log2 as _r,log10 as gr,log1p as yr,logaddexp as xr,logaddexp2 as hr,absolute as wr,abs as Ar,negative as vr,sign as Sr,mod as kr,divide as zr,true_divide as Nr,floor_divide as qr,positive as Ir,reciprocal as Dr,cbrt as Tr,fabs as Cr,divmod as Or,square as Pr,remainder as Er,heaviside as Lr,float_power as jr,fmod as Rr,frexp as Gr,gcd as Hr,lcm as Ur,ldexp as Yr,modf as Mr,clip as Vr,maximum as Br,minimum as Fr,fmax as Jr,fmin as Kr,nan_to_num as Qr,interp as Wr,unwrap as Xr,sinc as Zr,i0 as $r}from"./arithmetic.js";import{sin as ee,cos as ne,tan as ae,arcsin as te,asin as oe,arccos as me,acos as ie,arctan as ue,atan as se,arctan2 as be,atan2 as fe,hypot as pe,degrees as le,radians as de,deg2rad as ce,rad2deg as _e,sinh as ge,cosh as ye,tanh as xe,arcsinh as he,asinh as we,arccosh as Ae,acosh as ve,arctanh as Se,atanh as ke}from"./trig.js";import{dot as Ne,matmul as qe,trace as Ie,diagonal as De,kron as Te,transpose as Ce,inner as Oe,outer as Pe,tensordot as Ee,einsum as Le,einsum_path as je,vdot as Re,vecdot as Ge,matrix_transpose as He,permute_dims as Ue,matvec as Ye,vecmat as Me,cross as Ve,linalg as Be}from"./linalg.js";import{reshape as Je,flatten as Ke,ravel as Qe,squeeze as We,expand_dims as Xe,swapaxes as Ze,moveaxis as $e,rollaxis as rn,concatenate as en,stack as nn,vstack as an,hstack as tn,dstack as on,concat as mn,column_stack as un,row_stack as sn,block as bn,split as fn,array_split as pn,vsplit as ln,hsplit as dn,dsplit as cn,unstack as _n,tile as gn,repeat as yn,flip as xn,fliplr as hn,flipud as wn,rot90 as An,roll as vn,resize as Sn,atleast_1d as kn,atleast_2d as zn,atleast_3d as Nn}from"./shape.js";import{sum as In,mean as Dn,prod as Tn,max as Cn,amax as On,min as Pn,amin as En,ptp as Ln,argmin as jn,argmax as Rn,variance as Gn,var_ as Hn,variance as Un,std as Yn,median as Mn,percentile as Vn,quantile as Bn,average as Fn,all as Jn,any as Kn,cumsum as Qn,cumulative_sum as Wn,cumprod as Xn,cumulative_prod as Zn,nansum as $n,nanprod as ra,nanmean as ea,nanvar as na,nanstd as aa,nanmin as ta,nanmax as oa,nanargmin as ma,nanargmax as ia,nancumsum as ua,nancumprod as sa,nanmedian as ba,nanquantile as fa,nanpercentile as pa}from"./reduction.js";import{logical_and as da,logical_or as ca,logical_not as _a,logical_xor as ga,isfinite as ya,isinf as xa,isnan as ha,isnat as wa,isneginf as Aa,isposinf as va,iscomplex as Sa,iscomplexobj as ka,isreal as za,isrealobj as Na,real_if_close as qa,isfortran as Ia,isscalar as Da,iterable as Ta,isdtype as Ca,promote_types as Oa,copysign as Pa,signbit as Ea,nextafter as La,spacing as ja,greater as Ra,greater_equal as Ga,less as Ha,less_equal as Ua,equal as Ya,not_equal as Ma,isclose as Va,allclose as Ba}from"./logic.js";import{sort as Ja,argsort as Ka,lexsort as Qa,partition as Wa,argpartition as Xa,sort_complex as Za,nonzero as $a,argwhere as rt,flatnonzero as et,where as nt,searchsorted as at,extract as tt,count_nonzero as ot}from"./sorting.js";import{bitwise_and as it,bitwise_or as ut,bitwise_xor as st,bitwise_not as bt,invert as ft,left_shift as pt,right_shift as lt,packbits as dt,unpackbits as ct,bitwise_count as _t,bitwise_invert as gt,bitwise_left_shift as yt,bitwise_right_shift as xt}from"./bitwise.js";import{around as wt,round as At,ceil as vt,fix as St,floor as kt,rint as zt,trunc as Nt}from"./rounding.js";import{unique as It,in1d as Dt,intersect1d as Tt,isin as Ct,setdiff1d as Ot,setxor1d as Pt,union1d as Et,trim_zeros as Lt,unique_all as jt,unique_counts as Rt,unique_inverse as Gt,unique_values as Ht}from"./sets.js";import{bincount as Yt,digitize as Mt,histogram as Vt,histogram2d as Bt,histogramdd as Ft,correlate as Jt,convolve as Kt,cov as Qt,corrcoef as Wt,histogram_bin_edges as Xt,trapezoid as Zt}from"./statistics.js";import{diff as ro,ediff1d as eo,gradient as no}from"./gradient.js";import{real as to,imag as oo,conj as mo,conjugate as io,angle as uo}from"./complex.js";import{broadcast_to as bo,broadcast_arrays as fo,broadcast_shapes as po,take as lo,put as co,take_along_axis as _o,put_along_axis as go,choose as yo,compress as xo,select as ho,place as wo,putmask as Ao,copyto as vo,indices as So,ix_ as ko,ravel_multi_index as zo,unravel_index as No,diag_indices as qo,diag_indices_from as Io,fill_diagonal as Do,tril_indices as To,tril_indices_from as Co,triu_indices as Oo,triu_indices_from as Po,mask_indices as Eo,array_equal as Lo,array_equiv as jo,apply_along_axis as Ro,apply_over_axes as Go,may_share_memory as Ho,shares_memory as Uo,geterr as Yo,seterr as Mo,iindex as Vo,bindex as Bo,vindex as Fo}from"./advanced.js";import{set_printoptions as Ko,get_printoptions as Qo,printoptions as Wo,format_float_positional as Xo,format_float_scientific as Zo,base_repr as $o,binary_repr as rm,array2string as em,array_repr as nm,array_str as am}from"./formatting.js";import{ndim as om,shape as mm,size as im,item as um,tolist as sm,tobytes as bm,byteswap as fm,view as pm,tofile as lm,fill as dm}from"./utility.js";import{append as _m,delete_ as gm,delete_ as ym,insert as xm,pad as hm}from"./shape-extra.js";import{can_cast as Am,common_type as vm,result_type as Sm,min_scalar_type as km,issubdtype as zm,typename as Nm,mintypecode as qm}from"./typechecking.js";import{poly as Dm,polyadd as Tm,polyder as Cm,polydiv as Om,polyfit as Pm,polyint as Em,polymul as Lm,polysub as jm,polyval as Rm,roots as Gm}from"./polynomial.js";import{parseNpy as Um,parseNpyHeader as Ym,parseNpyData as Mm}from"../io/npy/parser.js";import{serializeNpy as Bm}from"../io/npy/serializer.js";import{UnsupportedDTypeError as Jm,InvalidNpyError as Km,SUPPORTED_DTYPES as Qm,DTYPE_TO_DESCR as Wm}from"../io/npy/format.js";import{parseNpz as Zm,parseNpzSync as $m,loadNpz as ri,loadNpzSync as ei}from"../io/npz/parser.js";import{serializeNpz as ai,serializeNpzSync as ti}from"../io/npz/serializer.js";import{parseTxt as mi,genfromtxt as ii,fromregex as ui}from"../io/txt/parser.js";import{serializeTxt as bi}from"../io/txt/serializer.js";import*as o from"../common/ops/random.js";import{NDArrayCore as m}from"../common/ndarray-core.js";function t(r){return r&&typeof r=="object"&&"_data"in r&&"_shape"in r?m.fromStorage(r):r}const g={seed:o.seed,get_state:o.get_state,set_state:o.set_state,get_bit_generator:o.get_bit_generator,set_bit_generator:o.set_bit_generator,default_rng:r=>{const e=o.default_rng(r),n=a=>a instanceof m?a.storage:a;return{random:a=>t(e.random(a)),integers:(a,s,b)=>t(e.integers(a,s,b)),standard_normal:a=>t(e.standard_normal(a)),normal:(a,s,b)=>t(e.normal(a,s,b)),uniform:(a,s,b)=>t(e.uniform(a,s,b)),choice:(a,s,b,f)=>t(e.choice(n(a),s,b,f instanceof m?f.storage:f)),permutation:a=>t(e.permutation(n(a))),shuffle:a=>e.shuffle(a instanceof m?a.storage:a),exponential:(a,s)=>t(e.exponential(a,s)),poisson:(a,s)=>t(e.poisson(a,s)),binomial:(a,s,b)=>t(e.binomial(a,s,b))}},Generator:o.Generator,random:r=>t(o.random(r)),rand:(...r)=>t(o.rand(...r)),randn:(...r)=>t(o.randn(...r)),randint:(r,e,n,a)=>t(o.randint(r,e,n,a)),random_sample:r=>t(o.random_sample(r)),ranf:r=>t(o.ranf(r)),sample:r=>t(o.sample(r)),random_integers:(r,e,n)=>t(o.random_integers(r,e,n)),bytes:o.bytes,uniform:(r,e,n)=>t(o.uniform(r,e,n)),normal:(r,e,n)=>t(o.normal(r,e,n)),standard_normal:r=>t(o.standard_normal(r)),exponential:(r,e)=>t(o.exponential(r,e)),standard_exponential:r=>t(o.standard_exponential(r)),gamma:(r,e,n)=>t(o.gamma(r,e,n)),standard_gamma:(r,e)=>t(o.standard_gamma(r,e)),beta:(r,e,n)=>t(o.beta(r,e,n)),chisquare:(r,e)=>t(o.chisquare(r,e)),noncentral_chisquare:(r,e,n)=>t(o.noncentral_chisquare(r,e,n)),f:(r,e,n)=>t(o.f(r,e,n)),noncentral_f:(r,e,n,a)=>t(o.noncentral_f(r,e,n,a)),standard_cauchy:r=>t(o.standard_cauchy(r)),standard_t:(r,e)=>t(o.standard_t(r,e)),laplace:(r,e,n)=>t(o.laplace(r,e,n)),logistic:(r,e,n)=>t(o.logistic(r,e,n)),lognormal:(r,e,n)=>t(o.lognormal(r,e,n)),gumbel:(r,e,n)=>t(o.gumbel(r,e,n)),pareto:(r,e)=>t(o.pareto(r,e)),power:(r,e)=>t(o.power(r,e)),rayleigh:(r,e)=>t(o.rayleigh(r,e)),triangular:(r,e,n,a)=>t(o.triangular(r,e,n,a)),wald:(r,e,n)=>t(o.wald(r,e,n)),weibull:(r,e)=>t(o.weibull(r,e)),poisson:(r,e)=>t(o.poisson(r,e)),binomial:(r,e,n)=>t(o.binomial(r,e,n)),geometric:(r,e)=>t(o.geometric(r,e)),hypergeometric:(r,e,n,a)=>t(o.hypergeometric(r,e,n,a)),logseries:(r,e)=>t(o.logseries(r,e)),negative_binomial:(r,e,n)=>t(o.negative_binomial(r,e,n)),zipf:(r,e)=>t(o.zipf(r,e)),multinomial:(r,e,n)=>t(o.multinomial(r,e,n)),multivariate_normal:(r,e,n,a,s)=>t(o.multivariate_normal(r,e,n,a,s)),dirichlet:(r,e)=>t(o.dirichlet(r,e)),vonmises:(r,e,n)=>t(o.vonmises(r,e,n)),choice:(r,e,n,a)=>t(o.choice(r,e,n,a)),permutation:r=>t(o.permutation(r)),shuffle:r=>o.shuffle(r instanceof m?r.storage:r)};import*as i from"../common/ops/fft.js";function u(r){return r instanceof m?r.storage:r}const y={fft:(r,e,n,a)=>m.fromStorage(i.fft(u(r),e,n,a)),ifft:(r,e,n,a)=>m.fromStorage(i.ifft(u(r),e,n,a)),fft2:(r,e,n,a)=>m.fromStorage(i.fft2(u(r),e,n,a)),ifft2:(r,e,n,a)=>m.fromStorage(i.ifft2(u(r),e,n,a)),fftn:(r,e,n,a)=>m.fromStorage(i.fftn(u(r),e,n,a)),ifftn:(r,e,n,a)=>m.fromStorage(i.ifftn(u(r),e,n,a)),rfft:(r,e,n,a)=>m.fromStorage(i.rfft(u(r),e,n,a)),irfft:(r,e,n,a)=>m.fromStorage(i.irfft(u(r),e,n,a)),rfft2:(r,e,n,a)=>m.fromStorage(i.rfft2(u(r),e,n,a)),irfft2:(r,e,n,a)=>m.fromStorage(i.irfft2(u(r),e,n,a)),rfftn:(r,e,n,a)=>m.fromStorage(i.rfftn(u(r),e,n,a)),irfftn:(r,e,n,a)=>m.fromStorage(i.irfftn(u(r),e,n,a)),hfft:(r,e,n,a)=>m.fromStorage(i.hfft(u(r),e,n,a)),ihfft:(r,e,n,a)=>m.fromStorage(i.ihfft(u(r),e,n,a)),fftfreq:(r,e)=>m.fromStorage(i.fftfreq(r,e)),rfftfreq:(r,e)=>m.fromStorage(i.rfftfreq(r,e)),fftshift:(r,e)=>m.fromStorage(i.fftshift(u(r),e)),ifftshift:(r,e)=>m.fromStorage(i.ifftshift(u(r),e))};export{v as ArrayStorage,A as Complex,Wm as DTYPE_TO_DESCR,Km as InvalidNpyError,w as NDArrayCore,Qm as SUPPORTED_DTYPES,Jm as UnsupportedDTypeError,Ar as abs,wr as absolute,ie as acos,ve as acosh,mr as add,Jn as all,Ba as allclose,On as amax,En as amin,uo as angle,Kn as any,_m as append,Ro as apply_along_axis,Go as apply_over_axes,D as arange,me as arccos,Ae as arccosh,te as arcsin,he as arcsinh,ue as arctan,be as arctan2,Se as arctanh,Rn as argmax,jn as argmin,Xa as argpartition,Ka as argsort,rt as argwhere,wt as around,I as array,em as array2string,Lo as array_equal,jo as array_equiv,nm as array_repr,pn as array_split,am as array_str,Y as asanyarray,L as asarray,B as asarray_chkfinite,M as ascontiguousarray,V as asfortranarray,oe as asin,we as asinh,se as atan,fe as atan2,ke as atanh,kn as atleast_1d,zn as atleast_2d,Nn as atleast_3d,Fn as average,$o as base_repr,rm as binary_repr,Yt as bincount,Bo as bindex,it as bitwise_and,_t as bitwise_count,gt as bitwise_invert,yt as bitwise_left_shift,bt as bitwise_not,ut as bitwise_or,xt as bitwise_right_shift,st as bitwise_xor,bn as block,fo as broadcast_arrays,po as broadcast_shapes,bo as broadcast_to,fm as byteswap,Am as can_cast,Tr as cbrt,vt as ceil,yo as choose,Vr as clip,un as column_stack,vm as common_type,xo as compress,mn as concat,en as concatenate,mo as conj,io as conjugate,Kt as convolve,U as copy,Pa as copysign,vo as copyto,Wt as corrcoef,Jt as correlate,ne as cos,ye as cosh,ot as count_nonzero,Qt as cov,Ve as cross,Xn as cumprod,Qn as cumsum,Zn as cumulative_prod,Wn as cumulative_sum,ce as deg2rad,le as degrees,ym as delete,gm as delete_,J as diag,qo as diag_indices,Io as diag_indices_from,K as diagflat,De as diagonal,ro as diff,Mt as digitize,zr as divide,Or as divmod,Ne as dot,cn as dsplit,on as dstack,eo as ediff1d,Le as einsum,je as einsum_path,N as empty,G as empty_like,Ya as equal,pr as exp,lr as exp2,Xe as expand_dims,dr as expm1,tt as extract,P as eye,Cr as fabs,y as fft,dm as fill,Do as fill_diagonal,St as fix,et as flatnonzero,Ke as flatten,xn as flip,hn as fliplr,wn as flipud,jr as float_power,kt as floor,qr as floor_divide,Jr as fmax,Kr as fmin,Rr as fmod,Xo as format_float_positional,Zo as format_float_scientific,Gr as frexp,$ as frombuffer,ar as fromfile,rr as fromfunction,er as fromiter,ui as fromregex,nr as fromstring,q as full,H as full_like,Hr as gcd,ii as genfromtxt,O as geomspace,Qo as get_printoptions,Yo as geterr,no as gradient,Ra as greater,Ga as greater_equal,Lr as heaviside,Vt as histogram,Bt as histogram2d,Xt as histogram_bin_edges,Ft as histogramdd,dn as hsplit,tn as hstack,pe as hypot,$r as i0,E as identity,Vo as iindex,oo as imag,Dt as in1d,So as indices,Oe as inner,xm as insert,Wr as interp,Tt as intersect1d,ft as invert,Va as isclose,Sa as iscomplex,ka as iscomplexobj,Ca as isdtype,ya as isfinite,Ia as isfortran,Ct as isin,xa as isinf,ha as isnan,wa as isnat,Aa as isneginf,va as isposinf,za as isreal,Na as isrealobj,Da as isscalar,zm as issubdtype,um as item,Ta as iterable,ko as ix_,Te as kron,Ur as lcm,Yr as ldexp,pt as left_shift,Ha as less,Ua as less_equal,Qa as lexsort,Be as linalg,T as linspace,ri as loadNpz,ei as loadNpzSync,cr as log,gr as log10,yr as log1p,_r as log2,xr as logaddexp,hr as logaddexp2,da as logical_and,_a as logical_not,ca as logical_or,ga as logical_xor,C as logspace,Eo as mask_indices,qe as matmul,He as matrix_transpose,Ye as matvec,Cn as max,Br as maximum,Ho as may_share_memory,Dn as mean,Mn as median,tr as meshgrid,Pn as min,km as min_scalar_type,Fr as minimum,qm as mintypecode,kr as mod,Mr as modf,$e as moveaxis,ur as multiply,Qr as nan_to_num,ia as nanargmax,ma as nanargmin,sa as nancumprod,ua as nancumsum,oa as nanmax,ea as nanmean,ba as nanmedian,ta as nanmin,pa as nanpercentile,ra as nanprod,fa as nanquantile,aa as nanstd,$n as nansum,na as nanvar,om as ndim,vr as negative,La as nextafter,$a as nonzero,Ma as not_equal,z as ones,R as ones_like,Pe as outer,dt as packbits,hm as pad,Um as parseNpy,Mm as parseNpyData,Ym as parseNpyHeader,Zm as parseNpz,$m as parseNpzSync,mi as parseTxt,Wa as partition,Vn as percentile,Ue as permute_dims,wo as place,Dm as poly,Tm as polyadd,Cm as polyder,Om as polydiv,Pm as polyfit,Em as polyint,Lm as polymul,jm as polysub,Rm as polyval,Ir as positive,fr as pow,br as power,Wo as printoptions,Tn as prod,Oa as promote_types,Ln as ptp,co as put,go as put_along_axis,Ao as putmask,Bn as quantile,_e as rad2deg,de as radians,g as random,Qe as ravel,zo as ravel_multi_index,to as real,qa as real_if_close,Dr as reciprocal,Er as remainder,yn as repeat,F as require,Je as reshape,Sn as resize,Sm as result_type,lt as right_shift,zt as rint,vn as roll,rn as rollaxis,Gm as roots,An as rot90,At as round,sn as row_stack,at as searchsorted,ho as select,Bm as serializeNpy,ai as serializeNpz,ti as serializeNpzSync,bi as serializeTxt,Ko as set_printoptions,Ot as setdiff1d,Mo as seterr,Pt as setxor1d,mm as shape,Uo as shares_memory,Sr as sign,Ea as signbit,ee as sin,Zr as sinc,ge as sinh,im as size,Ja as sort,Za as sort_complex,ja as spacing,fn as split,sr as sqrt,Pr as square,We as squeeze,nn as stack,Yn as std,ir as subtract,In as sum,Ze as swapaxes,lo as take,_o as take_along_axis,ae as tan,xe as tanh,Ee as tensordot,gn as tile,bm as tobytes,lm as tofile,sm as tolist,Ie as trace,Ce as transpose,Zt as trapezoid,Q as tri,W as tril,To as tril_indices,Co as tril_indices_from,Lt as trim_zeros,X as triu,Oo as triu_indices,Po as triu_indices_from,Nr as true_divide,Nt as trunc,Nm as typename,Et as union1d,It as unique,jt as unique_all,Rt as unique_counts,Gt as unique_inverse,Ht as unique_values,ct as unpackbits,No as unravel_index,_n as unstack,Xr as unwrap,Z as vander,Un as var,Hn as var_,Gn as variance,Re as vdot,Ge as vecdot,Me as vecmat,pm as view,Fo as vindex,ln as vsplit,an as vstack,nt as where,k as zeros,j as zeros_like};

package/dist/esm/core/linalg.js

@@ -1 +1 @@
-import*as u from"../common/ops/linalg.js";import{toStorage as n,fromStorage as a,fromStorageView as A,Complex as C,ArrayStorage as s}from"./types.js";function D(r,e){const o=u.dot(n(r),n(e));return typeof o=="number"||typeof o=="bigint"||o instanceof C?o:a(o)}function N(r,e=0,o=0,t=1){const y=u.trace(n(r),e,o,t);return y instanceof s?a(y):y}function b(r,e=0,o=0,t=1){return a(u.diagonal(n(r),e,o,t))}function w(r,e){return a(u.kron(n(r),n(e)))}function l(r,e){return A(u.transpose(n(r),e),r)}function c(r,e){const o=u.inner(n(r),n(e));return typeof o=="number"||typeof o=="bigint"||o instanceof C?o:a(o)}function p(r,e){return a(u.outer(n(r),n(e)))}function f(r,e,o=2){const t=u.tensordot(n(r),n(e),o);return typeof t=="number"||typeof t=="bigint"||t instanceof C?t:a(t)}function k(r,...e){const o=e.map(y=>n(y)),t=u.einsum(r,...o);return typeof t=="number"||typeof t=="bigint"||t instanceof C?t:a(t)}function q(r,...e){const o=e.map(t=>n(t));return u.einsum_path(r,...o)}function S(r,e){return u.vdot(n(r),n(e))}function g(r,e,o=-1){const t=u.vecdot(n(r),n(e),o);return typeof t=="number"||typeof t=="bigint"||t instanceof C?t:a(t)}function x(r){return A(u.matrix_transpose(n(r)),r)}function d(r,e){return A(u.permute_dims(n(r),e),r)}function O(r,e){return a(u.matvec(n(r),n(e)))}function L(r,e){return a(u.vecmat(n(r),n(e)))}function v(r,e,o=-1,t=-1,y=-1,i){const m=u.cross(n(r),n(e),o,t,y,i);return typeof m=="number"||typeof m=="bigint"||m instanceof C?m:a(m)}function U(r,e){return a(u.matmul(n(r),n(e)))}const V={matmul:(r,e)=>a(u.matmul(n(r),n(e))),dot:(r,e)=>D(r,e),det:r=>{const e=u.det(n(r));return typeof e=="number"?e:a(e)},inv:r=>a(u.inv(n(r))),solve:(r,e)=>a(u.solve(n(r),n(e))),lstsq:(r,e,o)=>{const t=u.lstsq(n(r),n(e),o);return{x:a(t.x),residuals:a(t.residuals),rank:t.rank,s:a(t.s)}},norm:(r,e,o,t)=>{const y=u.norm(n(r),e,o,t);return typeof y=="number"?y:a(y)},cond:(r,e)=>u.cond(n(r),e),matrix_rank:(r,e)=>u.matrix_rank(n(r),e),matrix_power:(r,e)=>a(u.matrix_power(n(r),e)),pinv:(r,e)=>a(u.pinv(n(r),e)),qr:(r,e)=>{const o=u.qr(n(r),e);return"h"in o&&"tau"in o?{h:a(o.h),tau:a(o.tau)}:"q"in o&&"r"in o?{q:a(o.q),r:a(o.r)}:a(o)},cholesky:(r,e)=>a(u.cholesky(n(r),e)),svd:(r,e,o)=>{const t=u.svd(n(r),e,o);return"u"in t&&"s"in t&&"vt"in t?{u:a(t.u),s:a(t.s),vt:a(t.vt)}:a(t)},eig:r=>{const e=u.eig(n(r));return{w:a(e.w),v:a(e.v)}},eigh:(r,e)=>{const o=u.eigh(n(r),e);return{w:a(o.w),v:a(o.v)}},eigvals:r=>a(u.eigvals(n(r))),eigvalsh:(r,e)=>a(u.eigvalsh(n(r),e)),slogdet:r=>{const e=u.slogdet(n(r)),o=e.sign instanceof s?a(e.sign):e.sign,t=e.logabsdet instanceof s?a(e.logabsdet):e.logabsdet;return{sign:o,logabsdet:t}},svdvals:r=>a(u.svdvals(n(r))),multi_dot:r=>a(u.multi_dot(r.map(e=>n(e)))),tensorinv:(r,e)=>a(u.tensorinv(n(r),e)),tensorsolve:(r,e,o)=>a(u.tensorsolve(n(r),n(e),o)),vector_norm:(r,e,o,t)=>{const y=u.vector_norm(n(r),e,o,t);return typeof y=="number"?y:a(y)},matrix_norm:(r,e,o)=>{const t=u.matrix_norm(n(r),e,o);return typeof t=="number"?t:a(t)},cross:(r,e,o,t,y,i)=>v(r,e,o,t,y,i),matrix_transpose:r=>x(r),permute_dims:(r,e)=>d(r,e),trace:(r,e,o,t)=>N(r,e,o,t),diagonal:(r,e,o,t)=>b(r,e,o,t),outer:(r,e)=>p(r,e),inner:(r,e)=>c(r,e),tensordot:(r,e,o)=>f(r,e,o),vecdot:(r,e,o)=>g(r,e,o),transpose:(r,e)=>l(r,e)};export{v as cross,b as diagonal,D as dot,k as einsum,q as einsum_path,c as inner,w as kron,V as linalg,U as matmul,x as matrix_transpose,O as matvec,p as outer,d as permute_dims,f as tensordot,N as trace,l as transpose,S as vdot,g as vecdot,L as vecmat};
+import*as u from"../common/ops/linalg.js";import{toStorage as n,fromStorage as a,fromStorageView as A,Complex as C,ArrayStorage as s}from"./types.js";function D(r,e){const o=u.dot(n(r),n(e));return typeof o=="number"||typeof o=="bigint"||o instanceof C?o:a(o)}function N(r,e=0,o=0,t=1){const y=u.trace(n(r),e,o,t);return y instanceof s?a(y):y}function b(r,e=0,o=0,t=1){return a(u.diagonal(n(r),e,o,t))}function w(r,e){return a(u.kron(n(r),n(e)))}function l(r,e){return A(u.transpose(n(r),e),r)}function c(r,e){const o=u.inner(n(r),n(e));return typeof o=="number"||typeof o=="bigint"||o instanceof C?o:a(o)}function p(r,e){return a(u.outer(n(r),n(e)))}function f(r,e,o=2){const t=u.tensordot(n(r),n(e),o);return typeof t=="number"||typeof t=="bigint"||t instanceof C?t:a(t)}function k(r,...e){const o=e.map(y=>n(y)),t=u.einsum(r,...o);return typeof t=="number"||typeof t=="bigint"||t instanceof C?t:a(t)}function q(r,...e){const o=e.map(t=>n(t));return u.einsum_path(r,...o)}function S(r,e){return u.vdot(n(r),n(e))}function g(r,e,o=-1){const t=u.vecdot(n(r),n(e),o);return typeof t=="number"||typeof t=="bigint"||t instanceof C?t:a(t)}function x(r){return A(u.matrix_transpose(n(r)),r)}function d(r,e){return A(u.permute_dims(n(r),e),r)}function O(r,e){return a(u.matvec(n(r),n(e)))}function L(r,e){return a(u.vecmat(n(r),n(e)))}function v(r,e,o=-1,t=-1,y=-1,i){const m=u.cross(n(r),n(e),o,t,y,i);return typeof m=="number"||typeof m=="bigint"||m instanceof C?m:a(m)}function U(r,e){return a(u.matmul(n(r),n(e)))}const V={matmul:(r,e)=>a(u.matmul(n(r),n(e))),dot:(r,e)=>D(r,e),det:r=>{const e=u.det(n(r));return typeof e=="number"||e instanceof C?e:a(e)},inv:r=>a(u.inv(n(r))),solve:(r,e)=>a(u.solve(n(r),n(e))),lstsq:(r,e,o)=>{const t=u.lstsq(n(r),n(e),o);return{x:a(t.x),residuals:a(t.residuals),rank:t.rank,s:a(t.s)}},norm:(r,e,o,t)=>{const y=u.norm(n(r),e,o,t);return typeof y=="number"?y:a(y)},cond:(r,e)=>u.cond(n(r),e),matrix_rank:(r,e)=>u.matrix_rank(n(r),e),matrix_power:(r,e)=>a(u.matrix_power(n(r),e)),pinv:(r,e)=>a(u.pinv(n(r),e)),qr:(r,e)=>{const o=u.qr(n(r),e);return"h"in o&&"tau"in o?{h:a(o.h),tau:a(o.tau)}:"q"in o&&"r"in o?{q:a(o.q),r:a(o.r)}:a(o)},cholesky:(r,e)=>a(u.cholesky(n(r),e)),svd:(r,e,o)=>{const t=u.svd(n(r),e,o);return"u"in t&&"s"in t&&"vt"in t?{u:a(t.u),s:a(t.s),vt:a(t.vt)}:a(t)},eig:r=>{const e=u.eig(n(r));return{w:a(e.w),v:a(e.v)}},eigh:(r,e)=>{const o=u.eigh(n(r),e);return{w:a(o.w),v:a(o.v)}},eigvals:r=>a(u.eigvals(n(r))),eigvalsh:(r,e)=>a(u.eigvalsh(n(r),e)),slogdet:r=>{const e=u.slogdet(n(r)),o=e.sign instanceof s?a(e.sign):e.sign,t=e.logabsdet instanceof s?a(e.logabsdet):e.logabsdet;return{sign:o,logabsdet:t}},svdvals:r=>a(u.svdvals(n(r))),multi_dot:r=>a(u.multi_dot(r.map(e=>n(e)))),tensorinv:(r,e)=>a(u.tensorinv(n(r),e)),tensorsolve:(r,e,o)=>a(u.tensorsolve(n(r),n(e),o)),vector_norm:(r,e,o,t)=>{const y=u.vector_norm(n(r),e,o,t);return typeof y=="number"?y:a(y)},matrix_norm:(r,e,o)=>{const t=u.matrix_norm(n(r),e,o);return typeof t=="number"?t:a(t)},cross:(r,e,o,t,y,i)=>v(r,e,o,t,y,i),matrix_transpose:r=>x(r),permute_dims:(r,e)=>d(r,e),trace:(r,e,o,t)=>N(r,e,o,t),diagonal:(r,e,o,t)=>b(r,e,o,t),outer:(r,e)=>p(r,e),inner:(r,e)=>c(r,e),tensordot:(r,e,o)=>f(r,e,o),vecdot:(r,e,o)=>g(r,e,o),transpose:(r,e)=>l(r,e)};export{v as cross,b as diagonal,D as dot,k as einsum,q as einsum_path,c as inner,w as kron,V as linalg,U as matmul,x as matrix_transpose,O as matvec,p as outer,d as permute_dims,f as tensordot,N as trace,l as transpose,S as vdot,g as vecdot,L as vecmat};

package/dist/esm/core/polynomial.js

@@ -1 +1 @@
-import{NDArrayCore as z}from"../common/ndarray-core.js";import{ArrayStorage as q}from"../common/storage.js";import{array as d}from"./creation.js";function D(n){return n instanceof z?n:d(n)}function I(n){const f=D(n),e=f.data,s=f.size;if(s===0)return d([1]);let r=[1];for(let u=0;u<s;u++){const i=e[u],a=new Array(r.length+1).fill(0);for(let l=0;l<r.length;l++)a[l]+=r[l],a[l+1]-=r[l]*i;r=a}return d(r)}function L(n,f){const e=D(n),s=D(f),r=e.data,u=s.data,i=Math.max(e.size,s.size),a=new Array(i).fill(0);for(let o=0;o<e.size;o++)a[i-e.size+o]+=r[o];for(let o=0;o<s.size;o++)a[i-s.size+o]+=u[o];let l=0;for(;l<a.length-1&&a[l]===0;)l++;return d(a.slice(l))}function S(n,f=1){let e=D(n);for(let s=0;s<f;s++){const r=e.data,u=e.size;if(u<=1)return d([0]);const i=[];for(let a=0;a<u-1;a++){const l=u-1-a;i.push(r[a]*l)}e=d(i)}return e}function T(n,f){const e=[...D(n).data],s=[...D(f).data];if(s.length===0||s.length===1&&s[0]===0)throw new Error("Division by zero polynomial");for(;e.length>1&&e[0]===0;)e.shift();for(;s.length>1&&s[0]===0;)s.shift();if(e.length<s.length)return[d([0]),d(e)];const r=[],u=[...e];for(;u.length>=s.length;){const i=u[0]/s[0];r.push(i);for(let a=0;a<s.length;a++)u[a]-=i*s[a];u.shift()}for(;u.length>1&&Math.abs(u[0])<1e-15;)u.shift();return[d(r.length>0?r:[0]),d(u.length>0?u:[0])]}function B(n,f,e){const s=n.data,r=f.data,u=n.size;if(e>=u)throw new Error("polyfit: degree must be less than number of points");const i=new Float64Array(u),a=new Float64Array(u);for(let t=0;t<u;t++)i[t]=Number(s[t]),a[t]=Number(r[t]);const l=[];for(let t=0;t<u;t++){const h=[];for(let b=e;b>=0;b--)h.push(Math.pow(i[t],b));l.push(h)}const o=[],y=[];for(let t=0;t<=e;t++){o.push([]);for(let b=0;b<=e;b++){let A=0;for(let p=0;p<u;p++)A+=l[p][t]*l[p][b];o[t].push(A)}let h=0;for(let b=0;b<u;b++)h+=l[b][t]*a[b];y.push(h)}const c=e+1,m=o.map((t,h)=>[...t,y[h]]);for(let t=0;t<c;t++){let h=t;for(let b=t+1;b<c;b++)Math.abs(m[b][t])>Math.abs(m[h][t])&&(h=b);[m[t],m[h]]=[m[h],m[t]];for(let b=t+1;b<c;b++){const A=m[b][t]/m[t][t];for(let p=t;p<=c;p++)m[b][p]-=A*m[t][p]}}const w=new Array(c).fill(0);for(let t=c-1;t>=0;t--){let h=m[t][c];for(let b=t+1;b<c;b++)h-=m[t][b]*w[b];w[t]=h/m[t][t]}return d(w)}function P(n,f=1,e=0){let s=D(n);const r=Array.isArray(e)?e:[e];for(let u=0;u<f;u++){const i=s.data,a=s.size,l=[];for(let y=0;y<a;y++){const c=a-y;l.push(i[y]/c)}const o=u<r.length?r[u]:0;l.push(o),s=d(l)}return s}function Q(n,f){const e=D(n),s=D(f),r=e.data,u=s.data,i=e.size+s.size-1,a=new Array(i).fill(0);for(let l=0;l<e.size;l++)for(let o=0;o<s.size;o++)a[l+o]+=r[l]*u[o];return d(a)}function V(n,f){const e=D(n),s=D(f),r=e.data,u=s.data,i=Math.max(e.size,s.size),a=new Array(i).fill(0);for(let o=0;o<e.size;o++)a[i-e.size+o]+=r[o];for(let o=0;o<s.size;o++)a[i-s.size+o]-=u[o];let l=0;for(;l<a.length-1&&a[l]===0;)l++;return d(a.slice(l))}function Z(n,f){const e=D(n),s=e.data;if(typeof f=="number"){let o=s[0];for(let y=1;y<e.size;y++)o=o*f+s[y];return o}const r=f instanceof z?f:d(f),u=r.data,i=r.size,a=e.size,l=new Float64Array(i);for(let o=0;o<i;o++){const y=Number(u[o]);let c=Number(s[0]);for(let m=1;m<a;m++)c=c*y+Number(s[m]);l[o]=c}return new z(q.fromData(l,Array.from(r.shape),"float64"))}function G(n){const e=[...D(n).data];for(;e.length>1&&e[0]===0;)e.shift();let s=0;for(;e.length>1&&e[e.length-1]===0;)e.pop(),s++;const r=e.length-1;if(r+s===0)return v([],[]);let i=[],a=[];if(r===1)i.push(-e[1]/e[0]),a.push(0);else if(r===2){const c=e[0],m=e[1],w=e[2],t=m*m-4*c*w;if(t>=0){const h=Math.sqrt(t);i.push((-m+h)/(2*c),(-m-h)/(2*c)),a.push(0,0)}else{const h=Math.sqrt(-t);i.push(-m/(2*c),-m/(2*c)),a.push(h/(2*c),-h/(2*c))}}else if(r>=3){const c=k(e,r);for(const m of c)i.push(m.re),a.push(m.im)}for(let c=0;c<s;c++)i.push(0),a.push(0);const l=i.map((c,m)=>m);l.sort((c,m)=>{const w=Math.sqrt(i[c]**2+a[c]**2),t=Math.sqrt(i[m]**2+a[m]**2);return Math.abs(w-t)>1e-10?t-w:Math.abs(i[c]-i[m])>1e-10?i[m]-i[c]:a[m]-a[c]});const o=l.map(c=>i[c]),y=l.map(c=>a[c]);return v(o,y)}function v(n,f){const e=n.length,s=new Float64Array(2*e);for(let u=0;u<e;u++)s[2*u]=n[u],s[2*u+1]=f[u];const r=q.fromData(s,[e],"complex128");return new z(r)}function k(n,f){const e=Array.from({length:f},()=>new Array(f).fill(0)),s=n[0];for(let r=0;r<f;r++)e[0][r]=-n[r+1]/s;for(let r=1;r<f;r++)e[r][r-1]=1;return R(e,f)}function R(n,f){const e=[];let r=f,u=0;const i=100*f;let a=f,l=0;for(;r>0&&u<i;){if(u++,r===a?l++:(l=0,a=r),r===1){e.push({re:n[0][0],im:0}),r=0;break}if(r===2){e.push(...j(n[0][0],n[0][1],n[1][0],n[1][1])),r=0;break}let o=r-1;for(;o>0;){const t=Math.abs(n[o-1][o-1])+Math.abs(n[o][o]),h=222e-18*(t===0?1:t);if(Math.abs(n[o][o-1])<=h){n[o][o-1]=0;break}o--}const y=r-o;if(y===1){e.push({re:n[r-1][r-1],im:0}),r--;continue}if(y===2){e.push(...j(n[r-2][r-2],n[r-2][r-1],n[r-1][r-2],n[r-1][r-1])),r-=2;continue}let c;if(l>0&&l%10===0)c=Math.abs(n[r-1][r-2])+Math.abs(n[r-2][r-3]);else{const t=n[r-2][r-2],h=n[r-2][r-1],b=n[r-1][r-2],A=n[r-1][r-1],p=t+A,N=t*A-h*b,g=p*p-4*N;if(g>=0){const C=Math.sqrt(g),M=(p+C)/2,x=(p-C)/2;c=Math.abs(M-A)<Math.abs(x-A)?M:x}else c=A}for(let t=o;t<r;t++)n[t][t]=n[t][t]-c;const m=[],w=[];for(let t=o;t<r-1;t++){const h=Math.hypot(n[t][t],n[t+1][t]),b=h===0?1:n[t][t]/h,A=h===0?0:n[t+1][t]/h;m.push(b),w.push(A);for(let p=t;p<r;p++){const N=n[t][p],g=n[t+1][p];n[t][p]=b*N+A*g,n[t+1][p]=-A*N+b*g}}for(let t=0;t<m.length;t++){const h=o+t,b=m[t],A=w[t],p=Math.min(h+2,r-1);for(let N=o;N<=p;N++){const g=n[N][h],C=n[N][h+1];n[N][h]=b*g+A*C,n[N][h+1]=-A*g+b*C}}for(let t=o;t<r;t++)n[t][t]=n[t][t]+c}if(r>0)for(let o=0;o<r;o++)e.push({re:n[o][o],im:0});return e}function j(n,f,e,s){const r=n+s,u=n*s-f*e,i=r*r-4*u;if(i>=0){const a=Math.sqrt(i);return[{re:(r+a)/2,im:0},{re:(r-a)/2,im:0}]}else{const a=Math.sqrt(-i);return[{re:r/2,im:a/2},{re:r/2,im:-a/2}]}}export{I as poly,L as polyadd,S as polyder,T as polydiv,B as polyfit,P as polyint,Q as polymul,V as polysub,Z as polyval,G as roots};
+import{NDArrayCore as T}from"../common/ndarray-core.js";import{ArrayStorage as q}from"../common/storage.js";import{isBigIntDType as F}from"../common/dtype.js";import{array as d}from"./creation.js";function C(e){return e instanceof T?e:d(e)}function R(e,i){const a=e[i];return typeof a=="bigint"?Number(a):a}function j(e){return e==="complex128"||e==="complex64"}function w(e){const i=e.data,a=e.size;if(j(e.dtype)){const t=[];for(let l=0;l<a;l++)t.push(Number(i[2*l]));return t}const s=[];for(let t=0;t<a;t++)s.push(R(i,t));return s}function _(e){if(e==="bool")throw new TypeError("numpy boolean subtract, the `-` operator, is not supported, use the bitwise_xor, the `^` operator, or the logical_xor function instead.")}function z(e){return e==="complex128"||e==="complex64"?"complex128":e==="float64"||e==="float32"||e==="float16"||e==="uint64"?"float64":"int64"}function E(e){return e==="complex128"||e==="complex64"?"complex128":"float64"}function B(e){return e==="complex128"||e==="complex64"?e:e==="float32"?"float32":e==="float16"?"float16":"float64"}function Z(e){const i=C(e),a=i.size,s=i.dtype==="float32"||i.dtype==="complex64"?"float32":"float64";if(a===0)return d([1],s);const t=w(i);let l=[1];for(let f=0;f<a;f++){const c=t[f],n=new Array(l.length+1).fill(0);for(let r=0;r<l.length;r++)n[r]+=l[r],n[r+1]-=l[r]*c;l=n}return d(l,s)}function G(e,i){const a=C(e),s=C(i),t=w(a),l=w(s),f=Math.max(t.length,l.length),c=new Array(f).fill(0);for(let r=0;r<t.length;r++)c[f-t.length+r]+=t[r];for(let r=0;r<l.length;r++)c[f-l.length+r]+=l[r];let n=0;for(;n<c.length-1&&c[n]===0;)n++;return d(c.slice(n),a.dtype)}function J(e,i=1){let a=C(e);const s=z(a.dtype);for(let t=0;t<i;t++){const l=w(a),f=l.length;if(f<=1)return d([0],s);const c=[];for(let n=0;n<f-1;n++){const r=f-1-n;c.push(l[n]*r)}a=d(c,s)}return a}function K(e,i){const a=C(e),s=C(i),t=w(a),l=w(s),f=B(a.dtype);if(l.length===0||l.length===1&&l[0]===0)throw new Error("Division by zero polynomial");for(;t.length>1&&t[0]===0;)t.shift();for(;l.length>1&&l[0]===0;)l.shift();if(t.length<l.length)return[d([0],f),d(t,f)];const c=[],n=[...t];for(;n.length>=l.length;){const r=n[0]/l[0];c.push(r);for(let g=0;g<l.length;g++)n[g]-=r*l[g];n.shift()}for(;n.length>1&&Math.abs(n[0])<1e-15;)n.shift();return[d(c.length>0?c:[0],f),d(n.length>0?n:[0],f)]}function U(e,i,a){if(e.dtype==="float16")throw new TypeError("array type float16 is unsupported in linalg");const s=e.size;if(a>=s)throw new Error("polyfit: degree must be less than number of points");const t=w(e),l=w(i),f=new Float64Array(s),c=new Float64Array(s);for(let o=0;o<s;o++)f[o]=t[o],c[o]=l[o];const n=[];for(let o=0;o<s;o++){const h=[];for(let m=a;m>=0;m--)h.push(Math.pow(f[o],m));n.push(h)}const r=[],g=[];for(let o=0;o<=a;o++){r.push([]);for(let m=0;m<=a;m++){let A=0;for(let D=0;D<s;D++)A+=n[D][o]*n[D][m];r[o].push(A)}let h=0;for(let m=0;m<s;m++)h+=n[m][o]*c[m];g.push(h)}const b=a+1,p=r.map((o,h)=>[...o,g[h]]);for(let o=0;o<b;o++){let h=o;for(let m=o+1;m<b;m++)Math.abs(p[m][o])>Math.abs(p[h][o])&&(h=m);[p[o],p[h]]=[p[h],p[o]];for(let m=o+1;m<b;m++){const A=p[m][o]/p[o][o];for(let D=o;D<=b;D++)p[m][D]-=A*p[o][D]}}const y=new Array(b).fill(0);for(let o=b-1;o>=0;o--){let h=p[o][b];for(let m=o+1;m<b;m++)h-=p[o][m]*y[m];y[o]=h/p[o][o]}const u=e.dtype==="complex128"||e.dtype==="complex64"?"complex128":"float64";return d(y,u)}function W(e,i=1,a=0){let s=C(e);const t=E(s.dtype),l=Array.isArray(a)?a:[a];for(let f=0;f<i;f++){const c=w(s),n=c.length,r=[];for(let b=0;b<n;b++){const p=n-b;r.push(c[b]/p)}const g=f<l.length?l[f]:0;r.push(g),s=d(r,t)}return s}function X(e,i){const a=C(e),s=C(i),t=w(a),l=w(s),f=t.length+l.length-1,c=new Array(f).fill(0);for(let n=0;n<t.length;n++)for(let r=0;r<l.length;r++)c[n+r]+=t[n]*l[r];return d(c,a.dtype)}function Y(e,i){const a=C(e),s=C(i);_(a.dtype);const t=w(a),l=w(s),f=Math.max(t.length,l.length),c=new Array(f).fill(0);for(let r=0;r<t.length;r++)c[f-t.length+r]+=t[r];for(let r=0;r<l.length;r++)c[f-l.length+r]-=l[r];let n=0;for(;n<c.length-1&&c[n]===0;)n++;return d(c.slice(n),a.dtype)}function $(e,i){const a=C(e),s=w(a);if(typeof i=="number"){let y=s[0];for(let u=1;u<s.length;u++)y=y*i+s[u];return y}const t=i instanceof T?i:d(i),l=w(t),f=t.size,c=s.length,n=a.dtype,r=q.empty(Array.from(t.shape),n),g=r.data,b=j(n),p=F(n);for(let y=0;y<f;y++){const u=l[y];let o=s[0];for(let h=1;h<c;h++)o=o*u+s[h];n==="bool"&&(o=o?1:0),b?(g[2*y]=o,g[2*y+1]=0):p?g[y]=BigInt(Math.round(o)):g[y]=o}return new T(r)}function H(e){const i=C(e);if(i.dtype==="float16")throw new TypeError("array type float16 is unsupported in linalg");const a=i.dtype==="float32"||i.dtype==="complex64"?"complex64":"complex128",s=w(i);for(;s.length>1&&s[0]===0;)s.shift();let t=0;for(;s.length>1&&s[s.length-1]===0;)s.pop(),t++;const l=s.length-1;if(l+t===0)return k([],[],a);let c=[],n=[];if(l===1)c.push(-s[1]/s[0]),n.push(0);else if(l===2){const p=s[0],y=s[1],u=s[2],o=y*y-4*p*u;if(o>=0){const h=Math.sqrt(o);c.push((-y+h)/(2*p),(-y-h)/(2*p)),n.push(0,0)}else{const h=Math.sqrt(-o);c.push(-y/(2*p),-y/(2*p)),n.push(h/(2*p),-h/(2*p))}}else if(l>=3){const p=S(s,l);for(const y of p)c.push(y.re),n.push(y.im)}for(let p=0;p<t;p++)c.push(0),n.push(0);const r=c.map((p,y)=>y);r.sort((p,y)=>{const u=Math.sqrt(c[p]**2+n[p]**2),o=Math.sqrt(c[y]**2+n[y]**2);return Math.abs(u-o)>1e-10?o-u:Math.abs(c[p]-c[y])>1e-10?c[y]-c[p]:n[y]-n[p]});const g=r.map(p=>c[p]),b=r.map(p=>n[p]);return k(g,b,a)}function k(e,i,a="complex128"){const s=e.length,t=q.empty([s],a),l=t.data;for(let f=0;f<s;f++)l[2*f]=e[f],l[2*f+1]=i[f];return new T(t)}function S(e,i){const a=Array.from({length:i},()=>new Array(i).fill(0)),s=e[0];for(let t=0;t<i;t++)a[0][t]=-e[t+1]/s;for(let t=1;t<i;t++)a[t][t-1]=1;return L(a,i)}function L(e,i){const a=[];let t=i,l=0;const f=100*i;let c=i,n=0;for(;t>0&&l<f;){if(l++,t===c?n++:(n=0,c=t),t===1){a.push({re:e[0][0],im:0}),t=0;break}if(t===2){a.push(...I(e[0][0],e[0][1],e[1][0],e[1][1])),t=0;break}let r=t-1;for(;r>0;){const u=Math.abs(e[r-1][r-1])+Math.abs(e[r][r]),o=222e-18*(u===0?1:u);if(Math.abs(e[r][r-1])<=o){e[r][r-1]=0;break}r--}const g=t-r;if(g===1){a.push({re:e[t-1][t-1],im:0}),t--;continue}if(g===2){a.push(...I(e[t-2][t-2],e[t-2][t-1],e[t-1][t-2],e[t-1][t-1])),t-=2;continue}let b;if(n>0&&n%10===0)b=Math.abs(e[t-1][t-2])+Math.abs(e[t-2][t-3]);else{const u=e[t-2][t-2],o=e[t-2][t-1],h=e[t-1][t-2],m=e[t-1][t-1],A=u+m,D=u*m-o*h,x=A*A-4*D;if(x>=0){const N=Math.sqrt(x),M=(A+N)/2,v=(A-N)/2;b=Math.abs(M-m)<Math.abs(v-m)?M:v}else b=m}for(let u=r;u<t;u++)e[u][u]=e[u][u]-b;const p=[],y=[];for(let u=r;u<t-1;u++){const o=Math.hypot(e[u][u],e[u+1][u]),h=o===0?1:e[u][u]/o,m=o===0?0:e[u+1][u]/o;p.push(h),y.push(m);for(let A=u;A<t;A++){const D=e[u][A],x=e[u+1][A];e[u][A]=h*D+m*x,e[u+1][A]=-m*D+h*x}}for(let u=0;u<p.length;u++){const o=r+u,h=p[u],m=y[u],A=Math.min(o+2,t-1);for(let D=r;D<=A;D++){const x=e[D][o],N=e[D][o+1];e[D][o]=h*x+m*N,e[D][o+1]=-m*x+h*N}}for(let u=r;u<t;u++)e[u][u]=e[u][u]+b}if(t>0)for(let r=0;r<t;r++)a.push({re:e[r][r],im:0});return a}function I(e,i,a,s){const t=e+s,l=e*s-i*a,f=t*t-4*l;if(f>=0){const c=Math.sqrt(f);return[{re:(t+c)/2,im:0},{re:(t-c)/2,im:0}]}else{const c=Math.sqrt(-f);return[{re:t/2,im:c/2},{re:t/2,im:-c/2}]}}export{Z as poly,G as polyadd,J as polyder,K as polydiv,U as polyfit,W as polyint,X as polymul,Y as polysub,$ as polyval,H as roots};

package/dist/esm/core/reduction.js

@@ -1 +1 @@
-import*as u from"../common/ops/reduction.js";import*as N from"../common/ops/shape.js";import{toStorage as b,fromStorage as y,Complex as A,ArrayStorage as p}from"./types.js";function f(o,r,n,e){const t=b(o).ndim,a=r.map(s=>s<0?t+s:s),m=[...a].sort((s,i)=>i-s),l=[...a].sort((s,i)=>s-i);let c=b(o);for(const s of m){const i=e(c,s,!1);if(!(i instanceof p)){if(n){const D=Array(t).fill(1),C=p.zeros(D,c.dtype);return C.iset(0,i),y(C)}return i}c=i}if(n)for(const s of l)c=N.expandDims(c,s);return y(c)}function _(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.sum(t,a,m));const e=u.sum(b(o),r,n);return typeof e=="number"||typeof e=="bigint"||e instanceof A?e:y(e)}function h(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.mean(t,a,m));const e=u.mean(b(o),r,n);return typeof e=="number"||e instanceof A?e:y(e)}function O(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.prod(t,a,m));const e=u.prod(b(o),r,n);return typeof e=="number"||typeof e=="bigint"||e instanceof A?e:y(e)}function x(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.max(t,a,m));const e=u.max(b(o),r,n);return typeof e=="number"||e instanceof A?e:y(e)}const z=x;function g(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.min(t,a,m));const e=u.min(b(o),r,n);return typeof e=="number"||e instanceof A?e:y(e)}const M=g;function j(o,r,n){const e=u.ptp(b(o),r,n);return typeof e=="number"||e instanceof A?e:y(e)}function w(o,r){const n=u.argmin(b(o),r);return typeof n=="number"?n:y(n)}function B(o,r){const n=u.argmax(b(o),r);return typeof n=="number"?n:y(n)}function v(o,r,n,e){if(Array.isArray(r))return f(o,r,e??!1,(a,m,l)=>u.variance(a,m,n,l));const t=u.variance(b(o),r,n,e);return typeof t=="number"?t:y(t)}const E=v;function F(o,r,n,e){if(Array.isArray(r))return f(o,r,e??!1,(a,m,l)=>u.std(a,m,n,l));const t=u.std(b(o),r,n,e);return typeof t=="number"?t:y(t)}function G(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.median(t,a,m));const e=u.median(b(o),r,n);return typeof e=="number"?e:y(e)}function H(o,r,n,e){const t=u.percentile(b(o),r,n,e);return typeof t=="number"?t:y(t)}function I(o,r,n,e){const t=u.quantile(b(o),r,n,e);return typeof t=="number"?t:y(t)}function J(o,r,n,e){const t=n?b(n):void 0,a=u.average(b(o),r,t,e);return typeof a=="number"||a instanceof A?a:y(a)}function K(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.all(t,a,m));const e=u.all(b(o),r,n);return typeof e=="boolean"?e:y(e)}function L(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.any(t,a,m));const e=u.any(b(o),r,n);return typeof e=="boolean"?e:y(e)}function S(o,r){return y(u.cumsum(b(o),r))}const P=S;function d(o,r){return y(u.cumprod(b(o),r))}const Q=d;function R(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.nansum(t,a,m));const e=u.nansum(b(o),r,n);return typeof e=="number"||e instanceof A?e:y(e)}function U(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.nanprod(t,a,m));const e=u.nanprod(b(o),r,n);return typeof e=="number"||e instanceof A?e:y(e)}function V(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.nanmean(t,a,m));const e=u.nanmean(b(o),r,n);return typeof e=="number"||e instanceof A?e:y(e)}function W(o,r,n,e){if(Array.isArray(r))return f(o,r,e??!1,(a,m,l)=>u.nanvar(a,m,n,l));const t=u.nanvar(b(o),r,n,e);return typeof t=="number"?t:y(t)}function X(o,r,n,e){if(Array.isArray(r))return f(o,r,e??!1,(a,m,l)=>u.nanstd(a,m,n,l));const t=u.nanstd(b(o),r,n,e);return typeof t=="number"?t:y(t)}function Y(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.nanmin(t,a,m));const e=u.nanmin(b(o),r,n);return typeof e=="number"||e instanceof A?e:y(e)}function Z(o,r,n){if(Array.isArray(r))return f(o,r,n??!1,(t,a,m)=>u.nanmax(t,a,m));const e=u.nanmax(b(o),r,n);return typeof e=="number"||e instanceof A?e:y(e)}function $(o,r){const n=u.nanargmin(b(o),r);return typeof n=="number"?n:y(n)}function k(o,r){const n=u.nanargmax(b(o),r);return typeof n=="number"?n:y(n)}function rr(o,r){return y(u.nancumsum(b(o),r))}function er(o,r){return y(u.nancumprod(b(o),r))}function nr(o,r,n){const e=u.nanmedian(b(o),r,n);return typeof e=="number"?e:y(e)}function or(o,r,n,e){const t=u.nanquantile(b(o),r,n,e);return typeof t=="number"?t:y(t)}function tr(o,r,n,e){const t=u.nanpercentile(b(o),r,n,e);return typeof t=="number"?t:y(t)}export{K as all,z as amax,M as amin,L as any,B as argmax,w as argmin,J as average,d as cumprod,S as cumsum,Q as cumulative_prod,P as cumulative_sum,x as max,h as mean,G as median,g as min,k as nanargmax,$ as nanargmin,er as nancumprod,rr as nancumsum,Z as nanmax,V as nanmean,nr as nanmedian,Y as nanmin,tr as nanpercentile,U as nanprod,or as nanquantile,X as nanstd,R as nansum,W as nanvar,H as percentile,O as prod,j as ptp,I as quantile,F as std,_ as sum,E as var_,v as variance};
+import*as u from"../common/ops/reduction.js";import*as N from"../common/ops/shape.js";import{toStorage as b,fromStorage as y,Complex as A,ArrayStorage as p}from"./types.js";function f(o,e,n,r){const t=b(o).ndim,a=e.map(s=>s<0?t+s:s),m=[...a].sort((s,i)=>i-s),l=[...a].sort((s,i)=>s-i);let c=b(o);for(const s of m){const i=r(c,s,!1);if(!(i instanceof p)){if(n){const D=Array(t).fill(1),C=p.zeros(D,c.dtype);return C.iset(0,i),y(C)}return i}c=i}if(n)for(const s of l)c=N.expandDims(c,s);return y(c)}function _(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.sum(t,a,m));const r=u.sum(b(o),e,n);return typeof r=="number"||typeof r=="bigint"||r instanceof A?r:y(r)}function h(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.mean(t,a,m));const r=u.mean(b(o),e,n);return typeof r=="number"||r instanceof A?r:y(r)}function j(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.prod(t,a,m));const r=u.prod(b(o),e,n);return typeof r=="number"||typeof r=="bigint"||r instanceof A?r:y(r)}function x(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.max(t,a,m));const r=u.max(b(o),e,n);return typeof r=="number"||r instanceof A?r:y(r)}const w=x;function g(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.min(t,a,m));const r=u.min(b(o),e,n);return typeof r=="number"||r instanceof A?r:y(r)}const O=g;function z(o,e,n){const r=u.ptp(b(o),e,n);return typeof r=="number"||r instanceof A?r:y(r)}function M(o,e){const n=u.argmin(b(o),e);return typeof n=="number"?n:y(n)}function B(o,e){const n=u.argmax(b(o),e);return typeof n=="number"?n:y(n)}function v(o,e,n,r){if(Array.isArray(e))return f(o,e,r??!1,(a,m,l)=>u.variance(a,m,n,l));const t=u.variance(b(o),e,n,r);return typeof t=="number"?t:y(t)}const E=v;function F(o,e,n,r){if(Array.isArray(e))return f(o,e,r??!1,(a,m,l)=>u.std(a,m,n,l));const t=u.std(b(o),e,n,r);return typeof t=="number"?t:y(t)}function G(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.median(t,a,m));const r=u.median(b(o),e,n);return typeof r=="number"||r&&typeof r=="object"&&"re"in r?r:y(r)}function H(o,e,n,r){const t=u.percentile(b(o),e,n,r);return typeof t=="number"?t:y(t)}function I(o,e,n,r){const t=u.quantile(b(o),e,n,r);return typeof t=="number"?t:y(t)}function J(o,e,n,r){const t=n?b(n):void 0,a=u.average(b(o),e,t,r);return typeof a=="number"||a instanceof A?a:y(a)}function K(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.all(t,a,m));const r=u.all(b(o),e,n);return typeof r=="boolean"?r:y(r)}function L(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.any(t,a,m));const r=u.any(b(o),e,n);return typeof r=="boolean"?r:y(r)}function S(o,e){return y(u.cumsum(b(o),e))}const P=S;function d(o,e){return y(u.cumprod(b(o),e))}const Q=d;function R(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.nansum(t,a,m));const r=u.nansum(b(o),e,n);return typeof r=="number"||r instanceof A?r:y(r)}function U(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.nanprod(t,a,m));const r=u.nanprod(b(o),e,n);return typeof r=="number"||r instanceof A?r:y(r)}function V(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.nanmean(t,a,m));const r=u.nanmean(b(o),e,n);return typeof r=="number"||r instanceof A?r:y(r)}function W(o,e,n,r){if(Array.isArray(e))return f(o,e,r??!1,(a,m,l)=>u.nanvar(a,m,n,l));const t=u.nanvar(b(o),e,n,r);return typeof t=="number"?t:y(t)}function X(o,e,n,r){if(Array.isArray(e))return f(o,e,r??!1,(a,m,l)=>u.nanstd(a,m,n,l));const t=u.nanstd(b(o),e,n,r);return typeof t=="number"?t:y(t)}function Y(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.nanmin(t,a,m));const r=u.nanmin(b(o),e,n);return typeof r=="number"||r instanceof A?r:y(r)}function Z(o,e,n){if(Array.isArray(e))return f(o,e,n??!1,(t,a,m)=>u.nanmax(t,a,m));const r=u.nanmax(b(o),e,n);return typeof r=="number"||r instanceof A?r:y(r)}function $(o,e){const n=u.nanargmin(b(o),e);return typeof n=="number"?n:y(n)}function k(o,e){const n=u.nanargmax(b(o),e);return typeof n=="number"?n:y(n)}function rr(o,e){return y(u.nancumsum(b(o),e))}function er(o,e){return y(u.nancumprod(b(o),e))}function nr(o,e,n){const r=u.nanmedian(b(o),e,n);return typeof r=="number"||r&&typeof r=="object"&&"re"in r?r:y(r)}function or(o,e,n,r){const t=u.nanquantile(b(o),e,n,r);return typeof t=="number"?t:y(t)}function tr(o,e,n,r){const t=u.nanpercentile(b(o),e,n,r);return typeof t=="number"?t:y(t)}export{K as all,w as amax,O as amin,L as any,B as argmax,M as argmin,J as average,d as cumprod,S as cumsum,Q as cumulative_prod,P as cumulative_sum,x as max,h as mean,G as median,g as min,k as nanargmax,$ as nanargmin,er as nancumprod,rr as nancumsum,Z as nanmax,V as nanmean,nr as nanmedian,Y as nanmin,tr as nanpercentile,U as nanprod,or as nanquantile,X as nanstd,R as nansum,W as nanvar,H as percentile,j as prod,z as ptp,I as quantile,F as std,_ as sum,E as var_,v as variance};

package/dist/esm/core/shape-extra.js

@@ -1 +1 @@
-import{NDArrayCore as I}from"../common/ndarray-core.js";import{array as z,zeros as N}from"./creation.js";import{concatenate as F,flatten as C}from"./shape.js";import{wasmPad2D as T}from"../common/wasm/pad.js";import{hasFloat16 as k}from"../common/dtype.js";function R(a,r,i){const d=r instanceof I?r:z(Array.isArray(r)?r:[r]);if(i===void 0){const n=C(a),S=C(d);return F([n,S],0)}return F([a,d],i)}function L(a,r,i){const d=Array.isArray(r)?r:[r],n=[...a.shape],S=a.storage;if(i===void 0){const t=C(a),m=t.storage,s=[],w=new Set(d.map(y=>y<0?t.size+y:y));for(let y=0;y<t.size;y++)w.has(y)||s.push(m.iget(y));return z(s,a.dtype)}const l=i<0?n.length+i:i,x=n[l],D=new Set(d.map(t=>t<0?x+t:t)),p=x-D.size,u=[...n];u[l]=p;const f=N(u,a.dtype),o=f.storage,c=[];let g=1;for(let t=n.length-1;t>=0;t--)c.unshift(g),g*=n[t];const e=[];g=1;for(let t=u.length-1;t>=0;t--)e.unshift(g),g*=u[t];const A=n.reduce((t,m)=>t*m,1);for(let t=0;t<A;t++){const m=[];let s=t;for(let h=0;h<n.length;h++)m.push(Math.floor(s/c[h])),s=s%c[h];if(D.has(m[l]))continue;let w=0;for(const h of D)h<m[l]&&w++;const y=[...m];y[l]-=w;let b=0;for(let h=0;h<u.length;h++)b+=y[h]*e[h];o.iset(b,S.iget(t))}return f}function W(a,r,i,d){const n=Array.isArray(r)?r:[r],S=i instanceof I?i:z(Array.isArray(i)?i:[i]);if(d===void 0){const l=C(a),x=C(S),D=l.data,p=x.data,u=Array.from(D);if(n.length===1){const f=n[0],o=f<0?l.size+f:f,c=Array.from(p);u.splice(o,0,...c)}else{const f=n.map((o,c)=>({idx:o<0?l.size+o:o,valIdx:c})).sort((o,c)=>o.idx-c.idx);for(let o=0;o<f.length;o++){const{idx:c,valIdx:g}=f[o],e=p[g%p.length];u.splice(c+o,0,e)}}return z(u,a.dtype)}throw new Error("insert along axis not fully implemented in standalone")}function $(a,r,i="constant",d=0){const n=[...a.shape],S=n.length;let l;typeof r=="number"?l=Array(S).fill([r,r]):Array.isArray(r)&&typeof r[0]=="number"?l=Array(S).fill(r):l=r;const x=n.map((e,A)=>e+l[A][0]+l[A][1]);if(i==="constant"&&d===0&&S===2&&typeof r=="number"){const e=T(a.storage,r);if(e)return new I(e)}if(i!=="constant")throw new Error(`pad mode '${i}' not fully implemented in standalone`);const D=N(x,a.dtype),p=D.storage;if(d!==0)for(let e=0;e<D.size;e++)p.iset(e,d);const u=a.storage,f=[];let o=1;for(let e=n.length-1;e>=0;e--)f.unshift(o),o*=n[e];const c=[];o=1;for(let e=x.length-1;e>=0;e--)c.unshift(o),o*=x[e];const g=n.reduce((e,A)=>e*A,1);if(a.dtype==="float16"&&k&&u.isCContiguous&&p.isCContiguous){const e=new Float32Array(u.data.subarray(u.offset,u.offset+g)),A=p.size,t=new Float32Array(A);d!==0&&t.fill(d);for(let m=0;m<g;m++){const s=[];let w=m;for(let b=0;b<n.length;b++)s.push(Math.floor(w/f[b])),w=w%f[b];let y=0;for(let b=0;b<x.length;b++)y+=(s[b]+l[b][0])*c[b];t[y]=e[m]}return p.data.set(t),D}for(let e=0;e<g;e++){const A=[];let t=e;for(let s=0;s<n.length;s++)A.push(Math.floor(t/f[s])),t=t%f[s];let m=0;for(let s=0;s<x.length;s++)m+=(A[s]+l[s][0])*c[s];p.iset(m,u.iget(e))}return D}export{R as append,L as delete,L as delete_,W as insert,$ as pad};
+import{NDArrayCore as z}from"../common/ndarray-core.js";import{ArrayStorage as k}from"../common/storage.js";import{Complex as I}from"../common/complex.js";import{array as N,zeros as F}from"./creation.js";import{concatenate as T,flatten as w}from"./shape.js";import{wasmPad2D as L}from"../common/wasm/pad.js";import{hasFloat16 as M,isComplexDType as P}from"../common/dtype.js";function G(u,a,c){const d=a instanceof z?a:N(Array.isArray(a)?a:[a]);if(c===void 0){const s=w(u),D=w(d);return T([s,D],0)}return T([u,d],c)}function v(u,a,c){const d=Array.isArray(a)?a:[a],s=[...u.shape],D=u.storage;if(c===void 0){const t=w(u),r=t.storage,n=[],p=new Set(d.map(l=>l<0?t.size+l:l));for(let l=0;l<t.size;l++)p.has(l)||n.push(r.iget(l));return N(n,u.dtype)}const m=c<0?s.length+c:c,g=s[m],x=new Set(d.map(t=>t<0?g+t:t)),S=g-x.size,b=[...s];b[m]=S;const h=F(b,u.dtype),C=h.storage,f=[];let o=1;for(let t=s.length-1;t>=0;t--)f.unshift(o),o*=s[t];const e=[];o=1;for(let t=b.length-1;t>=0;t--)e.unshift(o),o*=b[t];const y=s.reduce((t,r)=>t*r,1);for(let t=0;t<y;t++){const r=[];let n=t;for(let A=0;A<s.length;A++)r.push(Math.floor(n/f[A])),n=n%f[A];if(x.has(r[m]))continue;let p=0;for(const A of x)A<r[m]&&p++;const l=[...r];l[m]-=p;let i=0;for(let A=0;A<b.length;A++)i+=l[A]*e[A];C.iset(i,D.iget(t))}return h}function H(u,a,c,d){const s=Array.isArray(a)?a:[a],D=c instanceof z?c:N(Array.isArray(c)?c:[c]);if(d===void 0){const m=w(u),g=w(D),x=u.dtype;if(P(x)){const f=m.size,o=g.size,e=[];for(let r=0;r<f;r++)e.push(m.storage.iget(r));if(s.length===1){const r=s[0],n=r<0?f+r:r,p=[];for(let l=0;l<o;l++){const i=g.storage.iget(l);p.push(i instanceof I?i:new I(Number(i),0))}e.splice(n,0,...p)}else{const r=s.map((n,p)=>({idx:n<0?f+n:n,valIdx:p})).sort((n,p)=>n.idx-p.idx);for(let n=0;n<r.length;n++){const{idx:p,valIdx:l}=r[n],i=g.storage.iget(l%o),A=i instanceof I?i:new I(Number(i),0);e.splice(p+n,0,A)}}const y=k.empty([e.length],x),t=y.data;for(let r=0;r<e.length;r++)t[r*2]=e[r].re,t[r*2+1]=e[r].im;return new z(y)}const b=m.data,h=g.data,C=Array.from(b);if(s.length===1){const f=s[0],o=f<0?m.size+f:f,e=Array.from(h);C.splice(o,0,...e)}else{const f=s.map((o,e)=>({idx:o<0?m.size+o:o,valIdx:e})).sort((o,e)=>o.idx-e.idx);for(let o=0;o<f.length;o++){const{idx:e,valIdx:y}=f[o],t=h[y%h.length];C.splice(e+o,0,t)}}return N(C,x)}throw new Error("insert along axis not fully implemented in standalone")}function J(u,a,c="constant",d=0){const s=[...u.shape],D=s.length;let m;typeof a=="number"?m=Array(D).fill([a,a]):Array.isArray(a)&&typeof a[0]=="number"?m=Array(D).fill(a):m=a;const g=s.map((e,y)=>e+m[y][0]+m[y][1]);if(c==="constant"&&d===0&&D===2&&typeof a=="number"){const e=L(u.storage,a);if(e)return new z(e)}if(c!=="constant")throw new Error(`pad mode '${c}' not fully implemented in standalone`);const x=F(g,u.dtype),S=x.storage;if(d!==0)for(let e=0;e<x.size;e++)S.iset(e,d);const b=u.storage,h=[];let C=1;for(let e=s.length-1;e>=0;e--)h.unshift(C),C*=s[e];const f=[];C=1;for(let e=g.length-1;e>=0;e--)f.unshift(C),C*=g[e];const o=s.reduce((e,y)=>e*y,1);if(u.dtype==="float16"&&M&&b.isCContiguous&&S.isCContiguous){const e=new Float32Array(b.data.subarray(b.offset,b.offset+o)),y=S.size,t=new Float32Array(y);d!==0&&t.fill(d);for(let r=0;r<o;r++){const n=[];let p=r;for(let i=0;i<s.length;i++)n.push(Math.floor(p/h[i])),p=p%h[i];let l=0;for(let i=0;i<g.length;i++)l+=(n[i]+m[i][0])*f[i];t[l]=e[r]}return S.data.set(t),x}for(let e=0;e<o;e++){const y=[];let t=e;for(let n=0;n<s.length;n++)y.push(Math.floor(t/h[n])),t=t%h[n];let r=0;for(let n=0;n<g.length;n++)r+=(y[n]+m[n][0])*f[n];S.iset(r,b.iget(e))}return x}export{G as append,v as delete,v as delete_,H as insert,J as pad};

package/dist/esm/core/statistics.js

@@ -1 +1 @@
-import*as A from"../common/ops/statistics.js";import{NDArrayCore as d,toStorage as o,fromStorage as t}from"./types.js";function g(n,e,r){const a=e?o(e):void 0;return t(A.bincount(o(n),a,r))}function b(n,e,r){return t(A.digitize(o(n),o(e),r))}function f(n,e,r,a,u){const y=e instanceof d?o(e):e,i=u?o(u):void 0,m=A.histogram(o(n),y,r,a,i);return[t(m.hist),t(m.bin_edges)]}function l(n,e,r,a,u,y){let i;if(Array.isArray(r)&&r.length===2){const C=r[0]instanceof d?o(r[0]):r[0],D=r[1]instanceof d?o(r[1]):r[1];i=[C,D]}else i=r;const m=y?o(y):void 0,s=A.histogram2d(o(n),o(e),i,a,u,m);return[t(s.hist),t(s.x_edges),t(s.y_edges)]}function p(n,e,r,a,u){const y=u?o(u):void 0,i=A.histogramdd(o(n),e,r,a,y);return[t(i.hist),i.edges.map(m=>t(m))]}function S(n,e,r){return t(A.correlate(o(n),o(e),r))}function h(n,e,r){return t(A.convolve(o(n),o(e),r))}function x(n,e,r,a,u){return t(A.cov(o(n),e?o(e):void 0,r,a,u))}function v(n,e,r){return t(A.corrcoef(o(n),e?o(e):void 0,r))}function _(n,e,r,a){return t(A.histogram_bin_edges(o(n),e,r,a?o(a):void 0))}function z(n,e,r,a){const u=A.trapezoid(o(n),e?o(e):void 0,r,a);return typeof u=="number"?u:t(u)}export{g as bincount,h as convolve,v as corrcoef,S as correlate,x as cov,b as digitize,f as histogram,l as histogram2d,_ as histogram_bin_edges,p as histogramdd,z as trapezoid};
+import*as i from"../common/ops/statistics.js";import{Complex as D}from"../common/complex.js";import{NDArrayCore as C,toStorage as o,fromStorage as a}from"./types.js";function b(n,e,r){const u=e?o(e):void 0;return a(i.bincount(o(n),u,r))}function l(n,e,r){return a(i.digitize(o(n),o(e),r))}function p(n,e,r,u,t){const y=e instanceof C?o(e):e,A=t?o(t):void 0,m=i.histogram(o(n),y,r,u,A),s=m.hist;return[a(s),a(m.bin_edges)]}function S(n,e,r,u,t,y){let A;if(Array.isArray(r)&&r.length===2){const c=r[0]instanceof C?o(r[0]):r[0],d=r[1]instanceof C?o(r[1]):r[1];A=[c,d]}else A=r;const m=y?o(y):void 0,s=i.histogram2d(o(n),o(e),A,u,t,m);return[a(s.hist),a(s.x_edges),a(s.y_edges)]}function x(n,e,r,u,t){const y=t?o(t):void 0,A=i.histogramdd(o(n),e,r,u,y);return[a(A.hist),A.edges.map(m=>a(m))]}function h(n,e,r){return a(i.correlate(o(n),o(e),r))}function v(n,e,r){return a(i.convolve(o(n),o(e),r))}function _(n,e,r,u,t){return a(i.cov(o(n),e?o(e):void 0,r,u,t))}function z(n,e,r){return a(i.corrcoef(o(n),e?o(e):void 0,r))}function B(n,e,r,u){return a(i.histogram_bin_edges(o(n),e,r,u?o(u):void 0))}function q(n,e,r,u){const t=i.trapezoid(o(n),e?o(e):void 0,r,u);return typeof t=="number"||t instanceof D?t:a(t)}export{b as bincount,v as convolve,z as corrcoef,h as correlate,_ as cov,l as digitize,p as histogram,S as histogram2d,B as histogram_bin_edges,x as histogramdd,q as trapezoid};