launch_ref

Function launch_ref 

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

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