diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-11-03 13:10:28 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-11-03 13:10:28 +0100 |
| commit | b07b2350b656c6a790a97830dc43615953973c20 (patch) | |
| tree | 48bbd20dc68215b0893f438683b665d4ae0c9044 /candle-core/src/conv.rs | |
| parent | 1b5063f3ca223f387dd28ba3b69dadb88708905c (diff) | |
| download | candle-b07b2350b656c6a790a97830dc43615953973c20.tar.gz candle-b07b2350b656c6a790a97830dc43615953973c20.tar.bz2 candle-b07b2350b656c6a790a97830dc43615953973c20.zip | |
Test for the transposed conv1d. (#1254)
Diffstat (limited to 'candle-core/src/conv.rs')
| -rw-r--r-- | candle-core/src/conv.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/conv.rs b/candle-core/src/conv.rs index 710f6abf..fe923087 100644 --- a/candle-core/src/conv.rs +++ b/candle-core/src/conv.rs @@ -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})") } |