Simplify PiecewiseDiscontinuousExpand

jloveric · Jun 15, 2024 · 387f7a4 · 387f7a4
1 parent 0b6177a
commit 387f7a4
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/high_order_layers_torch/Basis.py b/high_order_layers_torch/Basis.py
@@ -283,9 +283,7 @@ def __call__(self, x):
         mat_trans[windex, wrange.view(-1)] = out.flatten()
         mat = mat_trans.reshape(
             mat.shape[0], mat.shape[1], mat.shape[2], mat.shape[3], mat.shape[4]
-        )
-
-        mat = mat.permute(4, 0, 1, 2, 3)
+        ).permute(4, 0, 1, 2, 3)
 
         return mat