cuquantum.cutensornet.tensor_qr¶

cuquantum.cutensornet.tensor_qr(intptr_t handle, intptr_t tensor_in, intptr_t raw_data_in, intptr_t tensor_q, intptr_t q, intptr_t tensor_r, intptr_t r, intptr_t workspace, intptr_t stream)[source]¶

Perform QR decomposition of a tensor.

Parameters

handle (intptr_t) – The library handle.
tensor_in (intptr_t) – The input tensor descriptor.
raw_data_in (intptr_t) – The pointer address (as Python int) to the input tensor (on device).
tensor_q (intptr_t) – The tensor descriptor for the output Q.
q (intptr_t) – The pointer address (as Python int) to the output tensor Q (on device).
tensor_r (intptr_t) – The tensor descriptor for the output R.
r (intptr_t) – The pointer address (as Python int) to the output tensor R (on device).
workspace (intptr_t) – The workspace descriptor.
stream (intptr_t) – The CUDA stream handle (cudaStream_t as Python int).