(&mut self, idx: usize, state: Tensor)
| 48 | } |
| 49 | |
| 50 | pub fn set(&mut self, idx: usize, state: Tensor) { |
| 51 | if idx >= self.states.len() { |
| 52 | self.states.resize(idx + 1, None); |
| 53 | } |
| 54 | self.states[idx] = Some(state); |
| 55 | } |
| 56 | |
| 57 | pub fn clear(&mut self) { |
| 58 | for s in &mut self.states { |
no outgoing calls