Skip to main content

read_tensor_validate

Function read_tensor_validate 

Source
pub fn read_tensor_validate<C: CubePrimitive + Default + IntoRuntime>(
    tensor: Tensor<C>,
    index: usize,
    unroll_factor: usize,
    kernel_name: String,
) -> C
Expand description

Returns the value at index in tensor within bounds.