From 7bd9b721c1e257bd9e29a6e29b5ca8734157e5e6 Mon Sep 17 00:00:00 2001
From: Chin-Yun Yu <lolimaster.cs03@nctu.edu.tw>
Date: Tue, 14 May 2024 22:15:50 +0100
Subject: [PATCH] fix: use reshape for noncontiguous dy

---
 src/libtorchaudio/lfilter.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)