Additional util method for Shape class (#2799)

sascha08-15 · Mar 29, 2018 · 44dd3de · 44dd3de
1 parent 8041327
commit 44dd3de
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/shape/Shape.java b/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/linalg/api/shape/Shape.java
@@ -2598,4 +2598,19 @@ public static boolean areShapesBroadcastable(@NonNull int[] x, @NonNull int[] y)
 
         return true;
     }
+
+
+    public static boolean hasDefaultStridesForShape(INDArray input){
+        if(!strideDescendingCAscendingF(input)){
+            return false;
+        }
+        char order = input.ordering();
+        int[] defaultStrides;
+        if(order == 'f'){
+            defaultStrides = ArrayUtil.calcStridesFortran(input.shape());
+        } else {
+            defaultStrides = ArrayUtil.calcStrides(input.shape());
+        }
+        return Arrays.equals(input.stride(), defaultStrides);
+    }
 }