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,
Available on crate feature api only.
Expand description

Applies a 3D convolution.

Accepts ConvOptions for symmetric padding, or PaddedConvOptions for asymmetric padding. An explicit pad operation handles asymmetric padding before dispatching to the convolution backend.