fix: use reshape for noncontiguous dy (#3794)

pytorch · Jun 24, 2024 · 7f6209b · 7f6209b
1 parent b829e93
commit 7f6209b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)