diff --git a/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxNDArray.java b/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxNDArray.java index 7d89d78c2d0b..67b3a2e395bf 100644 --- a/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxNDArray.java +++ b/mxnet/mxnet-engine/src/main/java/ai/djl/mxnet/engine/MxNDArray.java @@ -1527,6 +1527,8 @@ public NDArray oneHot(int depth, float onValue, float offValue, DataType dataTyp return manager.invoke("_npx_one_hot", this, params).toType(dataType, false); } + /** {@inheritDoc} */ + @Override public NDArray batchDot(NDArray other) { return manager.invoke("_npx_batch_dot", new NDArray[] {this, other}, null); } diff --git a/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArray.java b/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArray.java index ae5046c0b582..51abb4cd2873 100644 --- a/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArray.java +++ b/pytorch/pytorch-engine/src/main/java/ai/djl/pytorch/engine/PtNDArray.java @@ -1413,6 +1413,7 @@ public NDArray oneHot(int depth, float onValue, float offValue, DataType dataTyp throw new UnsupportedOperationException("Not implemented"); } + /** {@inheritDoc} */ @Override public NDArray batchDot(NDArray other) { throw new UnsupportedOperationException("Not implemented"); diff --git a/tensorflow/tensorflow-engine/src/main/java/ai/djl/tensorflow/engine/TfNDArray.java b/tensorflow/tensorflow-engine/src/main/java/ai/djl/tensorflow/engine/TfNDArray.java index d58f2df8a281..64e8fba5dcac 100644 --- a/tensorflow/tensorflow-engine/src/main/java/ai/djl/tensorflow/engine/TfNDArray.java +++ b/tensorflow/tensorflow-engine/src/main/java/ai/djl/tensorflow/engine/TfNDArray.java @@ -485,6 +485,7 @@ public NDArray oneHot(int depth, float onValue, float offValue, DataType dataTyp } } + /** {@inheritDoc} */ @Override public NDArray batchDot(NDArray other) { throw new UnsupportedOperationException("Not implemented");