cuquantum.cutensornet.tensor_qr

cuquantum.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.