From 151074bc29459ee6436937d9957129f8aa31396a Mon Sep 17 00:00:00 2001 From: Jaren Ashcraft Date: Fri, 19 Apr 2024 16:11:48 -0700 Subject: [PATCH] another broadcasted calculation fix --- katsu/mueller.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/katsu/mueller.py b/katsu/mueller.py index 8bd4d17..3aed65a 100644 --- a/katsu/mueller.py +++ b/katsu/mueller.py @@ -223,7 +223,7 @@ def decompose_diattenuator(M): diattenutation_norm = diattenuation_vector / D[..., np.newaxis] else: diattenutation_norm = diattenuation_vector / D - + # DD = diattenutation_norm @ np.swapaxes(diattenutation_norm,-2,-1) DD = broadcast_outer(diattenutation_norm, diattenutation_norm) @@ -232,6 +232,7 @@ def decompose_diattenuator(M): if M.ndim > 2: I = np.broadcast_to(I, [*M.shape[:-2], 3, 3]) + mD = mD[..., np.newaxis] inner_diattenuator = mD * I + (1 - mD) * DD # Eq. 19 Lu & Chipman