Skip to main content

launch_ref

cubecl_std::tensor::identity

Function launch_ref 

Source 
pub fn launch_ref(client: &Client, output: TensorBinding, dtype: ElemType)
Expand description

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