Struct PathExt

pub struct PathExt<'a> { /* private fields */ }

Implementations

impl<'a> PathExt<'a>

pub fn ones(&self, name: &str, dims: &[i64]) -> Tensor

Create a tensor variable initialized with ones.

pub fn sub<T: ToString>(&'a self, s: T) -> PathExt<'a>

Get a sub-path of the current path.

pub fn var( &self, name: &str, dims: &[i64], init: Init, ) -> Result<Tensor, TchError>

Create a tensor variable initialized with the given initializer.

pub fn var_copy(&self, name: &str, t: &Tensor) -> Tensor

Create a tensor variable initialized with the values from another tensor.

pub fn zeros(&self, name: &str, dims: &[i64]) -> Tensor

Create a tensor variable initialized with zeros.

Trait Implementations

impl<'a, T> Div<T> for &'a PathExt<'a>

where T: ToString,

type Output = PathExt<'a>

The resulting type after applying the / operator.

fn div(self, rhs: T) -> Self::Output

Performs the / operation.

impl<'a, T> Div<T> for &'a mut PathExt<'a>

where T: ToString,

type Output = PathExt<'a>

The resulting type after applying the / operator.

fn div(self, rhs: T) -> Self::Output

Performs the / operation.