Function burn_wgpu::kernel::reduce::init_reduce_output

source · 
pub fn init_reduce_output<E: WgpuElement, const D: usize>(
    input: &WgpuTensor<E, D>,
    reduce_dim: usize
) -> WgpuTensor<E, D>
Expand description

Creates an empty output tensor with reduce output shape