Skip to content

Commit 7f6209b

Browse files
authored
fix: use reshape for noncontiguous dy (#3794)
1 parent b829e93 commit 7f6209b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/libtorchaudio/lfilter.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ class DifferentiableIIR : public torch::autograd::Function<DifferentiableIIR> {
168168

169169
da = F::conv1d(
170170
dyda.view({1, n_batch * n_channel, -1}),
171-
dy.view({n_batch * n_channel, 1, -1}),
171+
dy.reshape({n_batch * n_channel, 1, -1}),
172172
F::Conv1dFuncOptions().groups(n_batch * n_channel))
173173
.view({n_batch, n_channel, -1})
174174
.sum(0)

0 commit comments

Comments
 (0)