Update mfstructure.swift (#52)

jjjkkkjjj · Mar 8, 2024 · 397fa33 · 397fa33
1 parent 423282b
commit 397fa33
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: