idx = np.hstack((np.arange(0,d), np.arange(2*d,4*d))).astype(int) 부분이 잘못된것 같습니다.

[todayfirst]

lrp.run에서

idx = np.hstack((np.arange(0,d), np.arange(2d,4d))).astype(int) # indices of gates i,f,o together
idx_i, idx_f, idx_c, idx_o = np.arange(0,d), np.arange(d,2d), np.arange(2d,3d), np.arange(3d,4*d) # indices of gates i,g,f,o separately

부분의 idx에 i,f,o의 인덱스만 저장하는 것 같은데 그럼
np.hstack((np.arange(0,2d), np.arange(3d,4*d))).astype(int)
로 수정 해야할 것 같습니다.

그래야 c의 인덱스인 2d 부터 3d 가 제외됩니다.