launch_ref

cubecl_std::tensor::identity

Function launch_ref 

Source 
pub fn launch_ref<R: Runtime>(
    client: &ComputeClient<R::Server>,
    output: &TensorHandleRef<'_, R>,
    dtype: StorageType,
)
Expand description

Launch identity matrix kernel by ref. Ensure output is a TensorHandleRef containing a square matrix. output will contain the identity matrix.