launch

cubecl_std::tensor::identity

Function launch 

Source 
pub fn launch<R: Runtime>(client: &ComputeClient<R>, output: &TensorHandle<R>)
Expand description

Launch identity matrix kernel. Ensure output is a TensorHandle containing a square matrix. output will contain the identity matrix.