[−][src]Struct tch::nn::BatchNorm2D
A batch-normalization layer.
Fields
running_mean: Tensor
running_var: Tensor
ws: Tensor
bs: Tensor
Methods
impl BatchNorm2D
[src]
pub fn new<'a, T: Borrow<Path<'a>>>(
vs: T,
out_dim: i64,
config: BatchNorm2DConfig
) -> BatchNorm2D
[src]
vs: T,
out_dim: i64,
config: BatchNorm2DConfig
) -> BatchNorm2D
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]
&self,
xs: &Tensor,
ys: &Tensor,
d: Device,
batch_size: i64
) -> f64
impl Debug for BatchNorm2D
[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]
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> Any for T where
T: 'static + ?Sized,
[src]
T: 'static + ?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>,