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]¶
对张量执行 QR 分解。
- 参数
handle (intptr_t) – 不透明句柄，持有 cuTensorNet 的库上下文。
desc_tensor_in (intptr_t) – 描述张量的模式、范围和其他元数据信息。
raw_data_in (intptr_t) – 指向输入张量的原始数据的指针（在设备内存中）。
desc_tensor_q (intptr_t) – 描述输出张量 Q 的模式、范围和其他元数据信息。
q (intptr_t) – 指向输出张量数据 Q 的指针（在设备内存中）。
desc_tensor_r (intptr_t) – 描述输出张量 R 的模式、范围和其他元数据信息。
r (intptr_t) – 指向输出张量数据 R 的指针（在设备内存中）。
work_desc (intptr_t) – 不透明结构，描述工作区。提供的工作区必须是
valid（工作区大小必须等于或大于最小需求）。请参阅workspace_get_memory_size()和workspace_set_memory()。stream (intptr_t) – 执行计算的 CUDA 流。