slice

burn_cubecl::kernel

Function slice 

Source 
pub fn slice<R: CubeRuntime>(
    tensor: CubeTensor<R>,
    indices: &[Range<usize>],
) -> CubeTensor<R>
Expand description

Slice a jit tensor with a set of ranges