提交 a2bb4dbf authored 作者: Virgile Andreani's avatar Virgile Andreani 提交者: Ricardo Vieira

Make non-strict zip strict in local_subtensor_of_alloc

上级 69037dbb
...@@ -683,7 +683,7 @@ def local_subtensor_of_alloc(fgraph, node): ...@@ -683,7 +683,7 @@ def local_subtensor_of_alloc(fgraph, node):
# Slices to take from val # Slices to take from val
val_slices = [] val_slices = []
for i, (sl, dim) in enumerate(zip(slices, dims, strict=False)): for i, (sl, dim) in enumerate(zip(slices, dims[: len(slices)], strict=True)):
# If val was not copied over that dim, # If val was not copied over that dim,
# we need to take the appropriate subtensor on it. # we need to take the appropriate subtensor on it.
if i >= n_added_dims: if i >= n_added_dims:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论