grid_sample

burn_cubecl::kernel::grid_sample

Function grid_sample 

Source 
pub fn grid_sample<R: CubeRuntime>(
    input: CubeTensor<R>,
    grid: CubeTensor<R>,
    options: GridSampleOptions,
) -> CubeTensor<R>
Expand description

Grid sample operation supporting bilinear interpolation