From 2a00ba9c21398c93c306df5a50f0c51bfc24b2fa Mon Sep 17 00:00:00 2001 From: Abe Levitan Date: Mon, 21 Jul 2025 17:00:43 +0200 Subject: [PATCH] Fix the initialization issue with nonuniform intensities --- src/cdtools/models/fancy_ptycho.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/cdtools/models/fancy_ptycho.py b/src/cdtools/models/fancy_ptycho.py index 4b3e2e2..c958105 100644 --- a/src/cdtools/models/fancy_ptycho.py +++ b/src/cdtools/models/fancy_ptycho.py @@ -407,7 +407,9 @@ class FancyPtycho(CDIModel): if hasattr(dataset, 'intensities') and dataset.intensities is not None: intensities = dataset.intensities.to(dtype=Ws.dtype)[:,...] weights = t.sqrt(intensities) - Ws *= (weights / t.mean(weights)) + Ws *= (weights / t.mean(weights)).reshape( + (len(weights),) + (1,)*(Ws.ndim - 1)) + if hasattr(dataset, 'mask') and dataset.mask is not None: mask = dataset.mask.to(t.bool)