Module identity

cubecl_std::tensor

Module identity

Functions§

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