pub fn max_ragged<T>(
input: &Array<T>,
ragged_len: &Array<u32>,
dim: i32
) -> (Array<T::InType>, Array<u32>) where
T: HasAfEnum,
T::InType: HasAfEnum,
Max reduction along given axis as per ragged lengths provided
input
contains the input values to be reducedragged_len
array containing number of elements to use when reducing along dim
dim
is the dimension along which the max operation occursTuple of Arrays:
input
along dim
according to ragged_len
input
along dim
according to ragged_len
use arrayfire::{Array, dim4, print, randu, max_ragged}; let vals: [f32; 6] = [1.0f32, 2.0, 3.0, 4.0, 5.0, 6.0]; let rlens: [u32; 2] = [9, 2]; let varr = Array::new(&vals, dim4![3, 2]); let rarr = Array::new(&rlens, dim4![1, 2]); print(&varr); print(&rarr); let (out, idx) = max_ragged(&varr, &rarr, 0); print(&out); print(&idx);
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4