Skip to main content

conv3d

Function conv3d 

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

Applies a 3D convolution.

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