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. |