Skip to main content

launch

cubecl_std::tensor::identity

Function launch 

Source 
pub fn launch(client: &Client, output: &TensorHandle)
Expand description

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