Type Alias ConvTranspose2D

pub type ConvTranspose2D = ConvTransposeND<[i64; 2]>;

A two dimension transposed convolution layer.

Aliased Type

struct ConvTranspose2D {
    pub ws: Tensor,
    pub bs: Option<Tensor>,
    /* private fields */
}

Fields

ws: Tensor

bs: Option<Tensor>

Trait Implementations

impl Module for ConvTranspose2D

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