Skip to content

P
pytensor
提交 a7be8923 authored 作者: Pascal Lamblin's avatar Pascal Lamblin
浏览文件
操作

Fix bug in local_subtensor_merge

The formula for getting an element from an index in a reversed sequence was wrong.
上级 fa84dc7e
隐藏空白字符变更
内嵌 并排
正在显示 包含 1 行增加 和 1 行删除
theano/tensor/opt.py
浏览文件 @ a7be8923
...@@ -1676,7 +1676,7 @@ def merge_two_slices(slice1, len1, slice2, len2): ...@@ -1676,7 +1676,7 @@ def merge_two_slices(slice1, len1, slice2, len2):
# case it was in reverse we need to realize that we do not want # case it was in reverse we need to realize that we do not want
# the k-th element from sl.start but the k-th element from # the k-th element from sl.start but the k-th element from
# sl.stop backwards # sl.stop backwards
n_val = sl1.stop - sl1.start - 1 - sl2 * sl1.step n_val = sl1.stop - 1 - sl2 * sl1.step
# we need to pick either n_val or p_val and then follow same # we need to pick either n_val or p_val and then follow same
# steps as above for covering the index error cases # steps as above for covering the index error cases
val = T.switch(T.lt(reverse1, 0), n_val, p_val) val = T.switch(T.lt(reverse1, 0), n_val, p_val)
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论