diff --git a/src/libtorchaudio/lfilter.cpp b/src/libtorchaudio/lfilter.cpp
index 0d0abf6cf3..e9b8d4eea2 100644
--- a/src/libtorchaudio/lfilter.cpp
+++ b/src/libtorchaudio/lfilter.cpp
@@ -168,7 +168,7 @@ class DifferentiableIIR : public torch::autograd::Function<DifferentiableIIR> {
 
       da = F::conv1d(
                dyda.view({1, n_batch * n_channel, -1}),
-               dy.view({n_batch * n_channel, 1, -1}),
+               dy.reshape({n_batch * n_channel, 1, -1}),
                F::Conv1dFuncOptions().groups(n_batch * n_channel))
                .view({n_batch, n_channel, -1})
                .sum(0)