Skip to content

Commit 4d9c5bb

Browse files
committed
feat: change back to use view
1 parent 2a59602 commit 4d9c5bb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fft_conv_pytorch/fft_conv.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def complex_matmul(a: Tensor, b: Tensor, groups: int = 1) -> Tensor:
2727
c = torch.zeros(real.shape, dtype=torch.complex64, device=a.device)
2828
c.real, c.imag = real, imag
2929

30-
return c.reshape(c.size(0), -1, *c.shape[3:])
30+
return c.view(c.size(0), -1, *c.shape[3:])
3131

3232

3333
def to_ntuple(val: Union[int, Iterable[int]], n: int) -> Tuple[int, ...]:

0 commit comments

Comments
 (0)