Turn tensor.Tensor.shape and tensor.Tensor.strides into tuples

src/ninetoothed/jit.py

@@ -178,7 +178,7 @@ def visit_Attribute(self, node):
             if isinstance(value, Tensor):
                 inner = value.dtype
 
-                return Symbol(inner.__dict__[node.attr]).node
+                return Symbol(getattr(inner, node.attr)).node
 
         self.generic_visit(node)
 

src/ninetoothed/tensor.py

@@ -24,8 +24,8 @@ def __init__(
         self.name = f"tensor_{type(self).num_instances}"
 
         if ndim is not None:
-            self.shape = [Symbol(self.size_string(i)) for i in range(ndim)]
-            self.strides = [Symbol(self.stride_string(i)) for i in range(ndim)]
+            self.shape = (Symbol(self.size_string(i)) for i in range(ndim))
+            self.strides = (Symbol(self.stride_string(i)) for i in range(ndim))
         else:
             self.shape = shape
 
@@ -191,6 +191,22 @@ def stride(self, dim=None):
 
         return self.strides[dim]
 
+    @property
+    def shape(self):
+        return self._shape
+
+    @shape.setter
+    def shape(self, value):
+        self._shape = tuple(value)
+
+    @property
+    def strides(self):
+        return self._strides
+
+    @strides.setter
+    def strides(self, value):
+        self._strides = tuple(value)
+
     @property
     def ndim(self):
         return len(self.shape)