Skip to content

Commit d26a833

Browse files
committed
Rename Tullio indices to match equations in docs
1 parent fc81d21 commit d26a833

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lrp/rules.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -440,7 +440,7 @@ for R in (ZeroRule, EpsilonRule, GammaRule, WSquareRule)
440440
modify_layer!(rule, layer)
441441
ãₖ = modify_input(rule, aₖ)
442442
zₖ₊₁ = modify_denominator(rule, preactivation(layer, ãₖ))
443-
@tullio Rₖ[j, b] = ãₖ[j, b] * layer.weight[k, j] * Rₖ₊₁[k, b] / zₖ₊₁[k, b]
443+
@tullio Rₖ[j, b] = layer.weight[i, j] * ãₖ[j, b] / zₖ₊₁[i, b] * Rₖ₊₁[i, b]
444444
reset!()
445445
return nothing
446446
end

0 commit comments

Comments
 (0)