diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-09-13 17:50:39 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-13 16:50:39 +0100 |
| commit | 9a465e1b2601195b65f1422f16793d6825252231 (patch) | |
| tree | 79b0b49e2fe25976b5de5259fc3cd6ebee5ffe88 /candle-core/src/op.rs | |
| parent | 31ab2ddaebe632f06f647c470259364eb53ca9e8 (diff) | |
| download | candle-9a465e1b2601195b65f1422f16793d6825252231.tar.gz candle-9a465e1b2601195b65f1422f16793d6825252231.tar.bz2 candle-9a465e1b2601195b65f1422f16793d6825252231.zip | |
Add 1d upsampling. (#839)
* Add 1d upsampling.
* Add the interpolate functions.
Diffstat (limited to 'candle-core/src/op.rs')
| -rw-r--r-- | candle-core/src/op.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-core/src/op.rs b/candle-core/src/op.rs index 9382b217..7940739c 100644 --- a/candle-core/src/op.rs +++ b/candle-core/src/op.rs @@ -116,6 +116,7 @@ pub enum Op { stride: (usize, usize), }, + UpsampleNearest1D(Tensor), UpsampleNearest2D(Tensor), Cat(Vec<Tensor>, usize), |