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.
1 parent 7b44445 commit 707c82eCopy full SHA for 707c82e
pytensor/graph/op.py
@@ -20,7 +20,6 @@
20
add_tag_trace,
21
get_variable_trace_string,
22
)
23
-from pytensor.tensor.utils import _gufunc_to_out_shape
24
25
26
if TYPE_CHECKING:
@@ -599,6 +598,8 @@ def inplace_on_inputs(self, allowed_inplace_inputs: list[int]) -> "Op":
599
598
600
def infer_shape(self, fgraph, node, input_shapes):
601
if hasattr(self, "gufunc_signature"):
+ from pytensor.tensor.utils import _gufunc_to_out_shape
602
+
603
return _gufunc_to_out_shape(self.gufunc_signature, input_shapes)
604
else:
605
raise NotImplementedError(f"Op {self} does not implement infer_shape")
0 commit comments