Skip to main content

grid_sample

burn_cubecl::kernel::grid_sample

Function grid_sample 

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

Grid sample operation supporting bilinear interpolation