提交 fffe37c0 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fix indexing issues

上级 2ef1058b
...@@ -1561,19 +1561,16 @@ class Scan(PureOp): ...@@ -1561,19 +1561,16 @@ class Scan(PureOp):
for idx in xrange(self.n_mit_mot + self.n_mit_sot): for idx in xrange(self.n_mit_mot + self.n_mit_sot):
mintap = numpy.min(self.tap_array[idx]) mintap = numpy.min(self.tap_array[idx])
maxtap = numpy.max(self.tap_array[idx]) maxtap = numpy.max(self.tap_array[idx])
if idx < self.n_mit_mot:
outmaxtap = numpy.max(self.mitmot_out_taps()[idx])
else:
outmaxtap = 0
seq = outs[idx] seq = outs[idx]
for k in self.tap_array[idx]: for k in self.tap_array[idx]:
if maxtap < 0: if outmaxtap -k != 0:
dim_offset = abs(maxtap) nw_seq = seq[k - mintap: -(outmaxtap-k)][::-1]
else:
dim_offset = 0
if maxtap == mintap and maxtap != 0:
nw_seq = seq[:abs(maxtap)]
elif maxtap - k != 0:
nw_seq = seq[dim_offset + k - mintap - 1:\
-(maxtap - k + 1)][::-1]
else: else:
nw_seq = seq[dim_offset + k - mintap - 1: -1][::-1] nw_seq = seq[k - mintap:][::-1]
outer_inp_seqs.append(nw_seq) outer_inp_seqs.append(nw_seq)
outer_inp_seqs += [ outer_inp_seqs += [
x[:-1][::-1] for x in self.outer_sitsot_outs(outs)] x[:-1][::-1] for x in self.outer_sitsot_outs(outs)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论