Add tensor.tolist() method (#125)

tripy/tests/frontend/test_tensor.py

@@ -203,3 +203,16 @@ def test_multiple_elements_boolean_fails(self):
 
         with pytest.raises(tp.TripyException):
             bool(tensor)
+
+    @pytest.mark.parametrize(
+        "tensor, expected",
+        [
+            (tp.Tensor([0]), [0]),
+            (tp.zeros((1, 1, 1)), [[[0]]]),
+            (tp.Tensor([[[0.1]]]), [[[0.1]]]),
+            (tp.Tensor([True]), [True]),
+            (tp.ones((1, 2), dtype=tp.float16), [[1.0, 1.0]]),
+        ],
+    )
+    def test_tolist(self, tensor, expected):
+        assert np.allclose(tensor.tolist(), expected)

tripy/tripy/frontend/tensor.py

@@ -220,6 +220,9 @@ def data(self) -> Array:
             arr = cast(Tensor(arr), tripy.common.datatype.float32).eval()
         return arr
 
+    def tolist(self) -> List:
+        return self.data().data()
+
     def __iter__(self):
         raise TypeError("Iterating over tensors is not supported")