Function tch::nn::conv

source · [−]

pub fn conv<'a, ND: AsRef<[i64]>, T: Borrow<Path<'a>>>(
    vs: T, 
    in_dim: i64, 
    out_dim: i64, 
    ksizes: ND, 
    config: ConvConfigND<ND>
) -> Conv<ND>

Expand description

Creates a new convolution layer for any number of dimensions.