Function pull_out_bits

ciphercore-base/src/ops/utils.rs:46–55

Panics if `x` is not an array.

(x: Node)

/// Panics if `x` is not an array.
pub fn pull_out_bits(x: Node) -> Result<Node> {
    let shape = x.get_type()?.get_dimensions();
    if shape.len() == 1 {
        Ok(x)
    } else {
        let mut axes_permutation = vec![shape.len() as u64 - 1];
        axes_permutation.extend(0..shape.len() as u64 - 1);
        Ok(x.permute_axes(axes_permutation)?)
    }
}

pub fn pull_out_bits_for_type(t: Type) -> Result<Type> {
    if !t.is_array() {