Q2

Author: moe18

questions/2_transpose-of-a-matrix/tinygrad/solution.py

@@ -3,8 +3,7 @@
 def transpose_matrix_tg(a) -> Tensor:
     """
     Transpose a 2D matrix `a` using tinygrad.
-    Inputs can be Python lists, NumPy arrays, or tinygrad Tensors.
+    Inputs are tinygrad Tensors.
     Returns a transposed Tensor.
     """
-    a_t = Tensor(a)
-    return a_t.transpose(0,1)
+    return a.T

questions/2_transpose-of-a-matrix/tinygrad/starter_code.py

@@ -1,12 +1,9 @@
 from tinygrad.tensor import Tensor
 
-def transpose_matrix_tg(a) -> Tensor:
+def transpose_matrix_tg(a:Tensor) -> Tensor:
     """
     Transpose a 2D matrix `a` using tinygrad.
-    Inputs can be Python lists, NumPy arrays, or tinygrad Tensors.
+    Inputs are tinygrad Tensors.
     Returns a transposed Tensor.
     """
-    # Convert to Tensor
-    a_t = Tensor(a)
-    # Your implementation here
     pass

questions/2_transpose-of-a-matrix/tinygrad/tests.json

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