pub fn conv2d(
input: &CudaTensor,
weight: &CudaTensor,
bias: Option<&CudaTensor>,
stride: (usize, usize),
padding: (usize, usize),
stream: &CudaStream,
) -> CudaResult<CudaTensor>Expand description
2D convolution using cuDNN