Skip to content

GitLab

Unverified Commit 7f6209b4 authored by Chin-Yun Yu's avatar Chin-Yun Yu Committed by GitHub
Browse files

fix: use reshape for noncontiguous dy (#3794)

parent b829e936
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/libtorchaudio/lfilter.cpp
View file @ 7f6209b4
...@@ -168,7 +168,7 @@ class DifferentiableIIR : public torch::autograd::Function<DifferentiableIIR> { ...@@ -168,7 +168,7 @@ class DifferentiableIIR : public torch::autograd::Function<DifferentiableIIR> {
da = F::conv1d( da = F::conv1d(
dyda.view({1, n_batch * n_channel, -1}), 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)) F::Conv1dFuncOptions().groups(n_batch * n_channel))
.view({n_batch, n_channel, -1}) .view({n_batch, n_channel, -1})
.sum(0) .sum(0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment