Function validate_tensor_ndim 

pub fn validate_tensor_ndim(
    actual_ndim: usize,
    expected_ndim: usize,
    op_name: &str,
) -> PyResult<()>

Validate that input tensor has expected number of dimensions