huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 20:09:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add clone to various nn layers. (#910)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-20 11:33:51 +01:00
committed by GitHub
parent f685b2231c
commit 7b1ddcff47
7 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-nn/src/rnn.rs
View File

@ -85,7 +85,7 @@ impl LSTMConfig {
///
/// <https://en.wikipedia.org/wiki/Long_short-term_memory>
#[allow(clippy::upper_case_acronyms, unused)]
#[derive(Debug)]
#[derive(Clone, Debug)]
pub struct LSTM {
w_ih: Tensor,
w_hh: Tensor,
@ -235,7 +235,7 @@ impl GRUConfig {
///
/// <https://en.wikipedia.org/wiki/Gated_recurrent_unit>
#[allow(clippy::upper_case_acronyms, unused)]
#[derive(Debug)]
#[derive(Clone, Debug)]
pub struct GRU {
w_ih: Tensor,
w_hh: Tensor,