Type Definition tch::nn::Conv2D

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

Two dimensions convolution layer.

Trait Implementations

impl Module for Conv2D

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