Function funspace::utils::array_resized_axis

pub fn array_resized_axis<A, S, D, T>(
    input: &ArrayBase<S, D>, 
    size: usize, 
    axis: usize
) -> Array<T, D> where
    T: Zero + Clone,
    S: Data<Elem = A>,
    D: Dimension, 

Returns a new array with same dimensionality but different size n along the specified axis.

Example

use funspace::utils::array_resized_axis;
let array = ndarray::Array2::<f64>::zeros((5, 3));
let resized: ndarray::Array2<f64> = array_resized_axis(&array, 2, 1);
assert!(resized == ndarray::Array2::zeros((5, 2)));