提交 9ee0ee7c authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Fix wrong type-hint in indices_from_subtensor

上级 a1a805c9
......@@ -85,7 +85,7 @@ invalid_tensor_types = (
def indices_from_subtensor(
op_indices: Iterable[ScalarConstant],
idx_list: Optional[List[Union[Type, slice, Variable]]],
) -> Union[slice, Variable]:
) -> Tuple[Union[slice, Variable], ...]:
"""Recreate the index tuple from which a ``*Subtensor**`` ``Op`` was created.
Parameters
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论