From eff89f2ed6ed7b854f966c7e1fcef0e64ead74ae Mon Sep 17 00:00:00 2001 From: manjirisharma Date: Wed, 21 Oct 2026 04:40:23 +0530 Subject: [PATCH] fix(docs): pass array to visualize_image_attr for original_image in Captum tutorial (#3963) --- beginner_source/introyt/captumyt.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/beginner_source/introyt/captumyt.py b/beginner_source/introyt/captumyt.py index d08052caecc..05e249604af 100644 --- a/beginner_source/introyt/captumyt.py +++ b/beginner_source/introyt/captumyt.py @@ -247,8 +247,9 @@ attributions_ig = integrated_gradients.attribute(input_img, target=pred_label_idx, n_steps=200) # Show the original image for comparison -_ = viz.visualize_image_attr(None, np.transpose(transformed_img.squeeze().cpu().detach().numpy(), (1,2,0)), - method="original_image", title="Original Image") +orig_img_np = np.transpose(transformed_img.squeeze().cpu().detach().numpy(), (1, 2, 0)) +_ = viz.visualize_image_attr(orig_img_np, orig_img_np, + method="original_image", title="Original Image") default_cmap = LinearSegmentedColormap.from_list('custom blue', [(0, '#ffffff'),