From 31a1075f4b4799a4922fa0f617ee982baa5baa81 Mon Sep 17 00:00:00 2001
From: shua <gpg@isthisa.email>
Date: Mon, 19 Aug 2024 09:06:17 +0200
Subject: onnx: implement LSTM op (#2268)

use candle-nn LSTM
---
 candle-nn/src/rnn.rs | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'candle-nn')

diff --git a/candle-nn/src/rnn.rs b/candle-nn/src/rnn.rs
index 30ad6ff5..5934af85 100644
--- a/candle-nn/src/rnn.rs
+++ b/candle-nn/src/rnn.rs
@@ -55,6 +55,10 @@ pub struct LSTMState {
 }
 
 impl LSTMState {
+    pub fn new(h: Tensor, c: Tensor) -> Self {
+        LSTMState { h, c }
+    }
+
     /// The hidden state vector, which is also the output of the LSTM.
     pub fn h(&self) -> &Tensor {
         &self.h
-- 
cgit v1.2.3