From 0d00dcb224e2d7d2e858f5c01ffe00642e5524ee Mon Sep 17 00:00:00 2001 From: KexinFeng Date: Mon, 3 Jul 2023 21:07:34 -0700 Subject: [PATCH] Fixes rrror message in X.dot(w) (#2688) Co-authored-by: KexinFeng Co-authored-by: Frank Liu --- .../src/main/java/ai/djl/pytorch/engine/PtNDArray.java | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArray.java b/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArray.java index 65e0fb1b59a..14195d17bd3 100644 --- a/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArray.java +++ b/engines/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArray.java @@ -1351,8 +1351,9 @@ public PtNDArray dot(NDArray other) { int otherDim = other.getShape().dimension(); if (selfDim != otherDim || selfDim > 2) { throw new UnsupportedOperationException( - "Dimension mismatch or high dimensional dot operation is not supported. Please" - + " use .matMul instead."); + "Dimension mismatch or dimension is greater than 2. Dot product is only" + + " applied on two 1D vectors. For high dimensions, please use .matMul" + + " instead."); } return JniUtils.dot(this, manager.from(other)); }