cuquantum.cutensornet.workspace_compute_qr_sizes¶
- cuquantum.cutensornet.workspace_compute_qr_sizes(intptr_t handle, intptr_t desc_tensor_in, intptr_t desc_tensor_q, intptr_t desc_tensor_r, intptr_t work_desc)[源代码]¶
计算执行张量 QR 操作所需的工作空间大小。
- 参数
handle (intptr_t) – 持有 cuTensorNet 库上下文的不透明句柄。
desc_tensor_in (intptr_t) – 描述张量的模式、范围和其他元数据信息。
desc_tensor_q (intptr_t) – 描述输出张量 Q 的模式、范围和其他元数据信息。
desc_tensor_r (intptr_t) – 描述输出张量 R 的模式、范围和其他元数据信息。
work_desc (intptr_t) – 在其中收集信息的工作区描述符。