From 9736236175633de60bbc174cadf23f25c37ce653 Mon Sep 17 00:00:00 2001
From: Lennard <lennard.koehler@gmx.net>
Date: Fri, 1 Sep 2023 09:08:41 +0200
Subject: [PATCH] Allow retrieving and setting prefix of VarBuilder (#699)

---
 candle-nn/src/var_builder.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/candle-nn/src/var_builder.rs b/candle-nn/src/var_builder.rs
index c372897a..03929681 100644
--- a/candle-nn/src/var_builder.rs
+++ b/candle-nn/src/var_builder.rs
@@ -102,6 +102,20 @@ impl<'a, B: Backend> VarBuilderArgs<'a, B> {
         }
     }
 
+    /// Returns the prefix of the `VarBuilder`.
+    pub fn prefix(&self) -> String {
+        self.path.join(".")
+    }
+
+    /// Returns a new `VarBuilder` with the prefix set to `prefix`.
+    pub fn set_prefix(&self, prefix: impl ToString) -> Self {
+        Self {
+            data: self.data.clone(),
+            path: vec![prefix.to_string()],
+            _phantom: std::marker::PhantomData,
+        }
+    }
+
     /// Return a new `VarBuilder` adding `s` to the current prefix. This can be think of as `cd`
     /// into a directory.
     pub fn push_prefix<S: ToString>(&self, s: S) -> Self {