Fixed error on LayerActivation documentation sample

pytorch · Jul 3, 2023 · 3f7a0b7 · 3f7a0b7
1 parent b1a9830
commit 3f7a0b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/captum/attr/_core/layer/layer_activation.py b/captum/attr/_core/layer/layer_activation.py
@@ -112,7 +112,7 @@ def attribute(
             >>> input = torch.randn(2, 3, 32, 32, requires_grad=True)
             >>> # Computes layer activation.
             >>> # attribution is layer output, with size Nx12x32x32
-            >>> attribution = layer_cond.attribute(input)
+            >>> attribution = layer_act.attribute(input)
         """
         with torch.no_grad():
             layer_eval = _forward_layer_eval(