Function check_shape 

pub fn check_shape(
    r: &TensorRef,
    expected: &[u32],
    op: &'static str,
) -> Result<(), TensorRefError>

Verify a TensorRef matches the expected shape; returns TensorRefError::ShapeMismatch on mismatch.