huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 19:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Test for the transposed conv1d. (#1254)

Browse Source
This commit is contained in:
Laurent Mazare
2023-11-03 13:10:28 +01:00
committed by GitHub
parent 1b5063f3ca
commit b07b2350b6
2 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/src/conv.rs
View File

@ -196,8 +196,8 @@ impl Tensor {
stride: usize,
dilation: usize,
) -> Result<Self> {
let (c_out, c_in_k, k_size) = kernel.dims3()?;
let (b_size, c_in, l_in) = self.dims3()?;
let (c_in_k, c_out, k_size) = kernel.dims3()?;
if c_in != c_in_k {
crate::bail!("in_channel mismatch between input ({c_in}) and kernel ({c_in_k})")
}