Type Alias Conv2D

pub type Conv2D = Conv<[i64; 2]>;

Two dimensions convolution layer.

Aliased Type

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

Fields

ws: Tensor

bs: Option<Tensor>

Trait Implementations

impl Module for Conv2D

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