[−][src]Struct tch::nn::VarStore
A VarStore is used to store variables used by one or multiple layers. It specifies a single device where all variables are stored.
Methods
impl VarStore
[src]
pub fn new(device: Device) -> VarStore
[src]
pub fn device(&self) -> Device
[src]
pub fn trainable_variables(&self) -> Vec<Tensor>
[src]
pub fn root(&self) -> Path
[src]
pub fn save(&self, path: &Path) -> Fallible<()>
[src]
pub fn load(&self, path: &Path) -> Fallible<()>
[src]
Auto Trait Implementations
Blanket Implementations
impl<T, U> Into for T where
U: From<T>,
[src]
U: From<T>,
impl<T> From for T
[src]
impl<T, U> TryFrom for T where
U: Into<T>,
[src]
U: Into<T>,
type Error = Infallible
The type returned in the event of a conversion error.
fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>
[src]
impl<T> Borrow for T where
T: ?Sized,
[src]
T: ?Sized,
impl<T> BorrowMut for T where
T: ?Sized,
[src]
T: ?Sized,
fn borrow_mut(&mut self) -> &mut T
[src]
impl<T, U> TryInto for T where
U: TryFrom<T>,
[src]
U: TryFrom<T>,
type Error = <U as TryFrom<T>>::Error
The type returned in the event of a conversion error.
fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>
[src]
impl<T> Any for T where
T: 'static + ?Sized,
[src]
T: 'static + ?Sized,