Skip to main content

unfold4d

burn::tensor::module

Function unfold4d 

pub fn unfold4d(
    x: Tensor<4>,
    kernel_size: [usize; 2],
    options: UnfoldOptions,
) -> Tensor<3>
Expand description

Applies a 4D to 3D unfold.