Function pull_out_bits_pair

ciphercore-base/src/ops/utils.rs:72–81

Like pull_out_bits but for a pair of nodes, aligning the "bits dimension" before pulling out.

(x: Node, y: Node)

// Like pull_out_bits but for a pair of nodes, aligning the "bits dimension" before pulling out.
pub fn pull_out_bits_pair(x: Node, y: Node) -> Result<(Node, Node)> {
    let num_dims = std::cmp::max(
        x.get_type()?.get_shape().len(),
        y.get_type()?.get_shape().len(),
    );
    Ok((
        pull_out_bits(reshape_prepending_dims(x, num_dims)?)?,
        pull_out_bits(reshape_prepending_dims(y, num_dims)?)?,
    ))
}

pub fn prepend_dims(shape: ArrayShape, num_dims: usize) -> Result<ArrayShape> {
    match shape.len() {