cuquantum.cutensornet.workspace_compute_svd_sizes¶
- cuquantum.cutensornet.workspace_compute_svd_sizes(intptr_t handle, intptr_t desc_tensor_in, intptr_t desc_tensor_u, intptr_t desc_tensor_v, intptr_t svd_config, intptr_t work_desc)[源代码]¶
计算执行张量 SVD 操作所需的工作区大小。
- 参数
handle (intptr_t) – 持有 cuTensorNet 库上下文的不透明句柄。
desc_tensor_in (intptr_t) – 描述张量的模式、范围和其他元数据信息。
desc_tensor_u (intptr_t) – 描述输出张量 U 的模式、范围和其他元数据信息。
desc_tensor_v (intptr_t) – 描述输出张量 V 的模式、范围和其他元数据信息。
svd_config (intptr_t) – 此数据结构保存用户请求的 svd 参数。
work_desc (intptr_t) – 在其中收集信息的工作区描述符。