Struct TransposeConv2d 

pub struct TransposeConv2d { /* private fields */ }

Transpose 2D Convolution layer (Deconvolution)

Implementations

impl TransposeConv2d

pub fn new( in_channels: usize, out_channels: usize, kernel_size: (usize, usize), stride: (usize, usize), padding: (usize, usize), output_padding: (usize, usize), ) -> Self

Trait Implementations

impl Module for TransposeConv2d

fn forward(&self, input: &Tensor) -> Tensor

Forward pass

fn parameters(&self) -> Vec<Tensor>

Get all trainable parameters

fn train(&mut self)

Set module to training mode

fn eval(&mut self)

Set module to evaluation mode

fn is_training(&self) -> bool

Check if in training mode

fn num_parameters(&self) -> usize

Number of trainable parameters