Add stack metthod

ricardoV94 · ricardoV94 · commit 6821b93cc868 · 2025-05-30T11:20:08.000+02:00
diff --git a/pytensor/xtensor/type.py b/pytensor/xtensor/type.py
@@ -306,6 +306,9 @@ def rename(self, new_name_or_name_dict=None, **names):
         new_out.name = new_name
         return new_out
 
+    def stack(self, dim, **dims):
+        return px.shape.stack(self, dim, **dims)
+
     # def swap_dims(self, *args, **kwargs):
     #     ...
     #

Original file line number	Diff line number	Diff line change
`@@ -306,6 +306,9 @@ def rename(self, new_name_or_name_dict=None, **names):`
`306`	`306`	`new_out.name = new_name`
`307`	`307`	`return new_out`
`308`	`308`
	`309`	`+ def stack(self, dim, **dims):`
	`310`	`+ return px.shape.stack(self, dim, **dims)`
	`311`	`+`
`309`	`312`	`# def swap_dims(self, args, *kwargs):`
`310`	`313`	`# ...`
`311`	`314`	`#`