Function axis_all 

pub fn axis_all<S, D, A>(
    arr: &ArrayBase<S, D>,
    axis: Axis,
) -> Result<Array1<bool>, LogicalError>
where
    A: Float,
    S: Data<Elem = A>,
    D: Dimension + RemoveAxis,

Tests whether all array elements along a given axis evaluates to True

Returns an array of booleans

Example:

use rust_optimal_transport as ot;
use ot::ndarray_logical::axis_all;
use ndarray::{prelude::*, Axis};

let arr = array![[f32::INFINITY, 42.], [2., 11.]];
assert_eq!(axis_all(&arr, Axis(0)).unwrap(), array![false, true]);