[−][src]Struct tch::nn::BatchNorm2D

pub struct BatchNorm2D { /* fields omitted */ }

A batch-normalization layer.

Methods

`impl BatchNorm2D`[src]

`pub fn new(vs: &Path, out_dim: i64, config: BatchNorm2DConfig) -> BatchNorm2D`[src]

Trait Implementations

`impl ModuleT for BatchNorm2D`[src]

`fn forward_t(&self, xs: &Tensor, train: bool) -> Tensor`[src]

`fn batch_accuracy_for_logits( &self, xs: &Tensor, ys: &Tensor, d: Device, batch_size: i64 ) -> f64`[src]

Auto Trait Implementations

`impl !Send for BatchNorm2D`

`impl !Sync for BatchNorm2D`

Blanket Implementations

`impl<T, U> Into for T where U: From<T>,` [src]

`fn into(self) -> U`[src]

`impl<T> From for T`[src]

`fn from(t: T) -> T`[src]

`impl<T, U> TryFrom for T where U: Into<T>,` [src]

`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]

`fn borrow(&self) -> &T`[src]

`impl<T> BorrowMut for T where T: ?Sized,` [src]

`fn borrow_mut(&mut self) -> &mut T`[src]

`impl<T, U> TryInto for T where U: TryFrom<T>,` [src]

`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]

`fn type_id(&self) -> TypeId`[src]

[−][src]Struct tch::nn::BatchNorm2D

Methods

impl BatchNorm2D[src]

pub fn new(vs: &Path, out_dim: i64, config: BatchNorm2DConfig) -> BatchNorm2D[src]

Trait Implementations

impl ModuleT for BatchNorm2D[src]

fn forward_t(&self, xs: &Tensor, train: bool) -> Tensor[src]

fn batch_accuracy_for_logits( &self, xs: &Tensor, ys: &Tensor, d: Device, batch_size: i64) -> f64[src]

Auto Trait Implementations

impl !Send for BatchNorm2D

impl !Sync for BatchNorm2D

Blanket Implementations

impl<T, U> Into for T where U: From<T>, [src]

fn into(self) -> U[src]

impl<T> From for T[src]

fn from(t: T) -> T[src]

impl<T, U> TryFrom for T where U: Into<T>, [src]

type Error = Infallible

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>[src]

impl<T> Borrow for T where T: ?Sized, [src]

fn borrow(&self) -> &T[src]

impl<T> BorrowMut for T where T: ?Sized, [src]

fn borrow_mut(&mut self) -> &mut T[src]

impl<T, U> TryInto for T where U: TryFrom<T>, [src]

type Error = <U as TryFrom<T>>::Error

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>[src]

impl<T> Any for T where T: 'static + ?Sized, [src]

fn type_id(&self) -> TypeId[src]

`impl BatchNorm2D`[src]

`pub fn new(vs: &Path, out_dim: i64, config: BatchNorm2DConfig) -> BatchNorm2D`[src]

`impl ModuleT for BatchNorm2D`[src]

`fn forward_t(&self, xs: &Tensor, train: bool) -> Tensor`[src]

`fn batch_accuracy_for_logits( &self, xs: &Tensor, ys: &Tensor, d: Device, batch_size: i64 ) -> f64`[src]

`impl !Send for BatchNorm2D`

`impl !Sync for BatchNorm2D`

`impl<T, U> Into for T where U: From<T>,` [src]

`fn into(self) -> U`[src]

`impl<T> From for T`[src]

`fn from(t: T) -> T`[src]

`impl<T, U> TryFrom for T where U: Into<T>,` [src]

`type Error = Infallible`

`fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>`[src]

`impl<T> Borrow for T where T: ?Sized,` [src]

`fn borrow(&self) -> &T`[src]

`impl<T> BorrowMut for T where T: ?Sized,` [src]

`fn borrow_mut(&mut self) -> &mut T`[src]

`impl<T, U> TryInto for T where U: TryFrom<T>,` [src]

`type Error = <U as TryFrom<T>>::Error`

`fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>`[src]

`impl<T> Any for T where T: 'static + ?Sized,` [src]

`fn type_id(&self) -> TypeId`[src]