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

Make non-strict zip strict in tensor/subtensor.py

上级 a2bb4dbf
......@@ -522,7 +522,7 @@ def basic_shape(shape, indices):
"""
res_shape = ()
for idx, n in zip(indices, shape, strict=False):
for n, idx in zip(shape[: len(indices)], indices, strict=True):
if isinstance(idx, slice):
res_shape += (slice_len(idx, n),)
elif isinstance(getattr(idx, "type", None), SliceType):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论