Merge remote-tracking branch 'refs/remotes/origin/master'

jjjkkkjjj · May 15, 2024 · 4f6b98e · 4f6b98e
2 parents 1f1e6dd + 397fa33
commit 4f6b98e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Sources/Matft/core/object/mfstructure.swift b/Sources/Matft/core/object/mfstructure.swift
@@ -79,7 +79,7 @@ internal func shape2strides(_ shape: inout [Int], mforder: MfOrder) -> [Int]{
     case .Row://, .None:
         var prevAxisNum = shape2size(&shape)
         for index in 0..<shape.count{
-            ret[index] = prevAxisNum / shape[index]
+            ret[index] = prevAxisNum / max(shape[index],1)
             prevAxisNum = ret[index]
         }
     case .Column: