Skip to main content

reduce_shape

Function reduce_shape 

Source
pub fn reduce_shape(
    input: &[i64],
    reduce_axes: &[usize],
    keepdim: bool,
) -> Vec<i64>
Expand description

§6.20-0007 reduce-family shape rule: the input shape with reduce_axes removed (keepdim=false) or set to 1 (keepdim=true) — derived from op semantics.