pub trait ArgLayerTraitConst: LayerTraitConst {
    fn as_raw_ArgLayer(&self) -> *const c_void;
}
Expand description

ArgMax/ArgMin layer

Note: returns indices as floats, which means the supported range is [-2^24; 2^24]

Required Methods§

Implementors§