We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
vectorize_graph
1 parent e25e8a2 commit 4facab6Copy full SHA for 4facab6
pytensor/graph/replace.py
@@ -232,13 +232,13 @@ def vectorize_graph(
232
def vectorize_graph(
233
outputs: Sequence[Variable],
234
replace: Mapping[Variable, Variable],
235
-) -> Sequence[Variable]: ...
+) -> list[Variable]: ...
236
237
238
239
outputs: Variable | Sequence[Variable],
240
241
-) -> Variable | Sequence[Variable]:
+) -> Variable | list[Variable]:
242
"""Vectorize outputs graph given mapping from old variables to expanded counterparts version.
243
244
Expanded dimensions must be on the left. Behavior is similar to the functional `numpy.vectorize`.
0 commit comments