add more type checks

microsoft · Sep 12, 2024 · 1f6dd04 · 1f6dd04
1 parent 4b76477
commit 1f6dd04
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/js/common/lib/tensor-impl.ts b/js/common/lib/tensor-impl.ts
@@ -221,7 +221,11 @@ export class Tensor implements TensorInterface {
           } else if (arg1 instanceof typedArrayConstructor) {
             data = arg1;
           } else if (arg1 instanceof Uint8ClampedArray) {
-            data = Uint8Array.from(arg1);
+            if (arg0 === 'uint8') {
+              data = Uint8Array.from(arg1);
+            } else {
+              throw new TypeError(`A Uint8ClampedArray tensor's data must be type of uint8`);
+            }
           } else {
             throw new TypeError(`A ${type} tensor's data must be type of ${typedArrayConstructor}`);
           }