mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 18:48:51 +00:00
Support for groups in conv-transpose1d. (#1731)
* Groups support in conv-transpose-1d. * Remove dangling file.
This commit is contained in:
@ -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)?;
|
||||
|
Reference in New Issue
Block a user