Super resolution network (pytorch#148)

ezyang · Sep 5, 2017 · d2563a9 · d2563a9
1 parent a783a37
commit d2563a9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/torch/autograd/_functions/tensor.py b/torch/autograd/_functions/tensor.py
@@ -176,6 +176,11 @@ def backward(ctx, grad_output):
 
 class Permute(Function):
 
+    @staticmethod
+    def primspec(g, input, dim_indices):
+        return (g.appendNode(g.create("Transpose", [input]))
+                .is_("axes", dim_indices))
+
     @staticmethod
     def forward(ctx, input, dim_indices):
         ctx.rev_dim_indices = [None for _ in range(len(dim_indices))]