Shape

deepjavalibrary · Jun 8, 2023 · 6c3d3e0 · 6c3d3e0
1 parent dab75c9
commit 6c3d3e0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/api/src/main/java/ai/djl/ndarray/types/Shape.java b/api/src/main/java/ai/djl/ndarray/types/Shape.java
@@ -135,6 +135,7 @@ public long[] getShape() {
      * @return the shape in the given dimension
      */
     public long get(int dimension) {
+        dimension = dimension + (dimension < 0 ? shape.length : 0);
         return shape[dimension];
     }