Skip to content

Commit 5b1e91d

Browse files
authored
docs: fix out of bounds size access (#1639)
1 parent bd9a734 commit 5b1e91d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/LSTMEncoderDecoder/main.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ function (rnn::RNNDecoder)((decoder_input, carry, target_len, target), ps, st)
177177
if target === nothing
178178
### This will be optimized out by Reactant
179179
target = similar(
180-
decoder_input, size(decoder_input, 1), target_len, size(decoder_input, 3)
180+
decoder_input, size(decoder_input, 1), target_len, size(decoder_input, 2)
181181
)
182182
fill!(target, 0)
183183
else

0 commit comments

Comments
 (0)