Skip to main content

validate_unique_axes

Function validate_unique_axes 

Source
pub fn validate_unique_axes(
    op: &'static str,
    role: &'static str,
    rank: usize,
    axes: &[usize],
) -> Result<()>
Expand description

Validate a subset of axes for a tensor rank.

ยงExamples

use tenferro_tensor::validate::validate_unique_axes;

validate_unique_axes("reduce_sum", "axis", 3, &[0, 2])?;
assert!(validate_unique_axes("reduce_sum", "axis", 2, &[2]).is_err());
assert!(validate_unique_axes("reduce_sum", "axis", 2, &[0, 0]).is_err());