tensor_qr#

cuquantum.bindings.cutensornet.tensor_qr( intptr_t handle, intptr_t desc_tensor_in, intptr_t raw_data_in, intptr_t desc_tensor_q, intptr_t q, intptr_t desc_tensor_r, intptr_t r, intptr_t work_desc, intptr_t stream, )[source]#

Performs QR decomposition of a tensor.

Parameters:

handle (intptr_t) – Opaque handle holding cuTensorNet’s library context.
desc_tensor_in (intptr_t) – Describes the modes, extents, and other metadata information of a tensor.
raw_data_in (intptr_t) – Pointer to the raw data of the input tensor (in device memory).
desc_tensor_q (intptr_t) – Describes the modes, extents, and other metadata information of the output tensor Q.
q (intptr_t) – Pointer to the output tensor data Q (in device memory).
desc_tensor_r (intptr_t) – Describes the modes, extents, and other metadata information of the output tensor R.
r (intptr_t) – Pointer to the output tensor data R (in device memory).
work_desc (intptr_t) – Opaque structure describing the workspace. The provided workspace must be valid (the workspace size must be the same as or larger than the minimum needed). See workspace_get_memory_size() & workspace_set_memory().
stream (intptr_t) – The CUDA stream on which the computation is performed.