[feat] finish tensor functions

Dioxane123 · Jun 29, 2023 · 64dc28b · 64dc28b
1 parent 43aa645
commit 64dc28b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/minitorch/tensor_functions.py b/minitorch/tensor_functions.py
@@ -210,7 +210,8 @@ class Permute(Function):
     @staticmethod
     def forward(ctx: Context, a: Tensor, order: Tensor) -> Tensor:
         ctx.save_for_backward(a.shape, a._tensor.strides)
-        return a._new(a._tensor.permute(*order))
+        ord = [int(order._tensor.get(i)) for i in order._tensor.indices()]
+        return a._new(a._tensor.permute(*ord))
 
     @staticmethod
     def backward(ctx: Context, grad_output: Tensor) -> Tuple[Tensor, float]: