Function caffe2_math::select
source · pub fn select<T, Context>(
n: i32,
d: i32,
x: *const T,
idx: *const i32,
y: *mut T,
context: *mut Context
)
Expand description
| Select does index selection of the rows | a N*D matrix x, and gives the N dimensional | vector y that contains the selected | data. |