Skip to main content

s

Macro s 

Source
macro_rules! s {
    ( $( $axis:expr ),* $(,)? ) => { ... };
}
Expand description

Build a SliceSpec from axis expressions.

use rlx_tensor::{ax, graph, rg, s, shape};

let g = graph("win", |g| {
    let x = g.input("x", shape![4, 16]);
    x.slice(s![ax(), rg(2, 10)])
});
assert_eq!(g.outputs.len(), 1);