Merge pull request #219 from adtzlr/fix-tovoigt

Fix `tovoigt()` helper
adtzlr · May 3, 2022 · 7c632b7 · 7c632b7
2 parents 053c44e + d4a127b
commit 7c632b7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -48,6 +48,9 @@ All notable changes to this project will be documented in this file. The format
 
 ## [Unreleased]
 
+### Fixed
+- Fix `tovoigt()` helper for data with more or less than two trailing axes.
+
 ## [3.1.0] - 2022-05-02
 
 ### Added

diff --git a/felupe/math/_tensor.py b/felupe/math/_tensor.py
@@ -279,8 +279,8 @@ def ddot(A, B, n=2, parallel=False):
 
 def tovoigt(A):
     "Convert (3, 3) tensor to (6, ) voigt notation."
-    B = np.zeros((6, *A.shape[-2:]))
+    B = np.zeros((6, *A.shape[2:]))
     ij = [(0, 0), (1, 1), (2, 2), (0, 1), (1, 2), (0, 2)]
     for i6, (i, j) in enumerate(ij):
-        B[i6, :, :] = A[i, j, :, :]
+        B[i6] = A[i, j]
     return B