Struct tch::nn::Path

pub struct Path<'a> { /* fields omitted */ }

Methods

impl<'a> Path<'a>

pub fn sub(&'a self, s: &str) -> Path<'a>

pub fn device(&self) -> Device

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

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

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

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

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

pub fn randn(&self, name: &str, dims: &[i64], mean: f64, stdev: f64) -> Tensor

pub fn uniform(&self, name: &str, dims: &[i64], lo: f64, up: f64) -> Tensor

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

Trait Implementations

impl<'a, '_> Div<&'_ str> for &'a mut Path<'a>

type Output = Path<'a>

The resulting type after applying the / operator.

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

Performs the / operation.