[FIX] Merge conflict

f-dangel · Dec 19, 2022 · 48e9fbe · 48e9fbe
1 parent a4c2ec9
commit 48e9fbe
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/backpack/utils/conv.py b/backpack/utils/conv.py
@@ -1,6 +1,7 @@
 """Utility functions for convolution layers."""
 
 from typing import Callable, Tuple, Type, Union
+from warnings import warn
 
 import torch
 from einops import rearrange
@@ -180,8 +181,6 @@ def unfold_by_conv(
         stride=module.stride,
     )
 
-    return unfold.reshape(N, C_in * kernel_size_numel, -1)
-
 
 def _grad_input_padding(
     grad_output, input_size, stride, padding, kernel_size, dilation=None