Skip to main content

conv3d

Function conv3d 

Source
pub fn conv3d(
    x: Tensor<5>,
    weight: Tensor<5>,
    bias: Option<Tensor<1>>,
    options: impl Into<PaddedConvOptions<3>>,
) -> Tensor<5>
Expand description

Applies a 3D convolution.

Accepts ConvOptions for symmetric padding, or PaddedConvOptions for asymmetric padding. Asymmetric 3D padding is not yet supported.