Add get_ids to GradStore (#2379)

2025-06-20 20:09:50 +00:00 · 2024-08-01 09:56:13 +01:00
parent 1ba87a9450
commit ce90287f45
1 changed files with 5 additions and 0 deletions
--- a/candle-core/src/backprop.rs
+++ b/candle-core/src/backprop.rs
@ -756,4 +756,9 @@ impl GradStore {
        };
        Ok(grad)
    }
+
+    /// Get the tensor ids of the stored gradient tensors
+    pub fn get_ids(&self) -> impl Iterator<Item = &TensorId> {
+        self.0.keys()
+    }
 }