fix bug in sequences

DoctorTeeth · Oct 5, 2015 · 5db7255 · 5db7255
1 parent 91b754a
commit 5db7255
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/util/sequences.py b/util/sequences.py
@@ -142,7 +142,7 @@ def associative_recall(seq_len, vec_size, item_size):
   inputs[b+2 + item_size] = fetch_bit
   # choose a random item to be the prompt
 
-  outputs[b+2+item_size+1:inputs.shape[0]] = items[idx+1]
+  outputs[-items[idx+1].shape[0]:inputs.shape[0]] = items[idx+1]
 
   return inputs, outputs, seq_len