Q3

Author: moe18

questions/3_reshape-matrix/tinygrad/solution.py

@@ -3,11 +3,10 @@
 def reshape_matrix_tg(a, new_shape) -> Tensor:
     """
     Reshape a 2D matrix `a` to shape `new_shape` using tinygrad.
-    Inputs can be Python lists, NumPy arrays, or tinygrad Tensors.
+    Inputs are tinygrad Tensors.
     Returns a Tensor of shape `new_shape`, or an empty Tensor on mismatch.
     """
     # Dimension check
     if len(a) * len(a[0]) != new_shape[0] * new_shape[1]:
         return Tensor([])
-    a_t = Tensor(a)
-    return a_t.reshape(new_shape)
+    return a.reshape(new_shape)

questions/3_reshape-matrix/tinygrad/starter_code.py

@@ -1,15 +1,9 @@
 from tinygrad.tensor import Tensor
 
-def reshape_matrix_tg(a, new_shape) -> Tensor:
+def reshape_matrix_tg(a:Tensor, new_shape:tuple) -> Tensor:
     """
     Reshape a 2D matrix `a` to shape `new_shape` using tinygrad.
-    Inputs can be Python lists, NumPy arrays, or tinygrad Tensors.
+    Inputs are tinygrad Tensors.
     Returns a Tensor of shape `new_shape`, or an empty Tensor on mismatch.
     """
-    # Dimension check
-    if len(a) * len(a[0]) != new_shape[0] * new_shape[1]:
-        return Tensor([])
-    # Convert to Tensor and reshape
-    a_t = Tensor(a)
-    # Your implementation here
     pass

questions/3_reshape-matrix/tinygrad/tests.json

@@ -1,10 +1,10 @@
 [
   {
-    "test": "from tinygrad.tensor import Tensor\nres = reshape_matrix_tg(\n    [[1,2,3],[4,5,6]],\n    (3, 2)\n)\nprint(res.numpy().tolist())",
+    "test": "from tinygrad.tensor import Tensor\nres = reshape_matrix_tg(\n    Tensor([[1,2,3],[4,5,6]]),\n    (3, 2)\n)\nprint(res.numpy().tolist())",
     "expected_output": "[[1, 2], [3, 4], [5, 6]]"
   },
   {
-    "test": "from tinygrad.tensor import Tensor\nres = reshape_matrix_tg(\n    [[1,2],[3,4]],\n    (3, 2)\n)\nprint(res.numpy().tolist())",
+    "test": "from tinygrad.tensor import Tensor\nres = reshape_matrix_tg(\n    Tensor([[1,2],[3,4]]),\n    (3, 2)\n)\nprint(res.numpy().tolist())",
     "expected_output": "[]"
   }
 ]