huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/tests/conv_tests.rs
View File

@ -50,7 +50,7 @@ fn conv1d(dev: &Device) -> Result<()> {
test_utils::to_vec1_round(&res.flatten_all()?, 4)?,
[2.4509, 2.6357, -1.3336, 4.1393, 0.5657, 1.8091, -1.1784, 3.5675, 0.5069, 3.3352]
);
let res = t.conv_transpose1d(&w.transpose(0, 1)?, 0, 0, 1, 1)?;
let res = t.conv_transpose1d(&w.transpose(0, 1)?, 0, 0, 1, 1, 1)?;
assert_eq!(res.dims(), [1, 2, 7]);
assert_eq!(
test_utils::to_vec1_round(&res.flatten_all()?, 4)?,