Function arrayfire::col

pub fn col(input: &Array, col_num: u64) -> Array

Extract col_num col from input Array

Examples

use arrayfire::{Dim4, randu, col, print};
let dims = Dim4::new(&[5, 5, 1, 1]);
let a = randu::<f32>(dims);
print(&a);
println!("Grab last col of the random matrix");
print(&col(&a, 4));