add methods

Author: OriolAbril

pytensor/xtensor/type.py

@@ -195,6 +195,15 @@ def rename(self, new_name_or_name_dict, **names):
             name_dict = new_name_or_name_dict
         return rename(name_dict, **names)
 
+
+    def stack(self, dim, **dims):
+        from pytensor.xtensor.shape import stack
+        return stack(self, dim, **dims)
+
+    def unstack(self, din, **dims):
+        from pytensor.xtensor.shape import unstack
+        return unstack(self, dim, **dims)
+
     # def swap_dims(self, *args, **kwargs):
     #     ...
     #