Module identity

Source

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.