Support for groups in conv-transpose1d. (#1731)

* Groups support in conv-transpose-1d.

* Remove dangling file.
This commit is contained in:
Laurent Mazare
2024-02-18 21:28:07 +01:00
committed by GitHub
parent cb86b0c82c
commit 1fb728772d
4 changed files with 61 additions and 27 deletions

View File

@ -250,6 +250,7 @@ impl Tensor {
out_padding,
*stride,
*dilation,
/* groups */ 1,
)?;
let sum_grad = grads.or_insert(arg)?;
*sum_grad = sum_grad.add(&grad_arg)?;