Skip to main content

conv3d

yscv_kernels

Function conv3d 

Source 
pub fn conv3d(
    input: &[f32],
    input_shape: &[usize],
    kernel: &[f32],
    kernel_shape: &[usize],
    stride: (usize, usize, usize),
    padding: (usize, usize, usize),
) -> (Vec<f32>, Vec<usize>)
Expand description

3D convolution: input [B, D, H, W, C_in], kernel [KD, KH, KW, C_in, C_out], output [B, OD, OH, OW, C_out] Supports padding and stride in all 3 dimensions.