conv2d

tch_plus::nn

Function 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.