Function tch::nn::conv2d

source · 
pub fn conv2d<'a, T: Borrow<Path<'a>>>(
    vs: T,
    i: i64,
    o: i64,
    k: i64,
    c: ConvConfig
) -> Conv2D
Expand description

Creates a new two dimension convolution layer.