Function put_in_bits

ciphercore-base/src/ops/utils.rs:105–114

Panics if `x` is not an array.

(x: Node)

/// Panics if `x` is not an array.
pub fn put_in_bits(x: Node) -> Result<Node> {
    let shape = x.get_type()?.get_dimensions();
    if shape.len() == 1 {
        Ok(x)
    } else {
        let mut axes_permutation: Vec<u64> = (1..shape.len()).map(|x| x as u64).collect();
        axes_permutation.push(0);
        Ok(x.permute_axes(axes_permutation)?)
    }
}

pub fn zeros_like(x: Node) -> Result<Node> {
    x.get_graph().zeros(x.get_type()?)