Type Definition tch::nn::Conv3D

type Conv3D = Conv<[i64; 3]>;

Three dimensions convolution layer.

Trait Implementations

impl Module for Conv3D

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