提交 0195a930 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

More precise type-hint for `vectorize_graph`

上级 46c23410
......@@ -232,13 +232,13 @@ def vectorize_graph(
def vectorize_graph(
outputs: Sequence[Variable],
replace: Mapping[Variable, Variable],
) -> Sequence[Variable]: ...
) -> list[Variable]: ...
def vectorize_graph(
outputs: Variable | Sequence[Variable],
replace: Mapping[Variable, Variable],
) -> Variable | Sequence[Variable]:
) -> Variable | list[Variable]:
"""Vectorize outputs graph given mapping from old variables to expanded counterparts version.
Expanded dimensions must be on the left. Behavior is similar to the functional `numpy.vectorize`.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论