From 5cac507c11284aa7d8f097835f44e39caf162ffb Mon Sep 17 00:00:00 2001
From: Alain Riou <alain.riou14000@yahoo.com>
Date: Fri, 1 Dec 2023 18:34:18 +0100
Subject: [PATCH] png issue

---
 CHANGELOG.md    | 10 ++++++++++
 pesto/export.py |  2 +-
 2 files changed, 11 insertions(+), 1 deletion(-)
 create mode 100644 CHANGELOG.md

diff --git a/CHANGELOG.md b/CHANGELOG.md
new file mode 100644
index 0000000..af27849
--- /dev/null
+++ b/CHANGELOG.md
@@ -0,0 +1,10 @@
+# Changelog
+
+## v0.1.1
+
+- solve issue when exporting in PNG
+
+
+## v0.1.0 - 2023-10-17
+
+Initial version
\ No newline at end of file
diff --git a/pesto/export.py b/pesto/export.py
index daab6a9..05c7c5a 100644
--- a/pesto/export.py
+++ b/pesto/export.py
@@ -41,7 +41,7 @@ def export_png(output_file: str, timesteps, confidence, activations, lims=(21, 1
 
     bps = activations.shape[1] // 128
     activations = activations[:, bps*lims[0]: bps*lims[1]]
-    activations = activations * confidence.unsqueeze(1)
+    activations = activations * confidence[:, None]
     plt.imshow(activations.T,
                aspect='auto', origin='lower', cmap='inferno',
                extent=(timesteps[0], timesteps[-1]) + lims)
-- 
GitLab