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)[source]

Computes the workspace size needed to perform the tensor SVD operation.

Parameters
  • handle (intptr_t) – Opaque handle holding cuTensorNet’s library context.

  • desc_tensor_in (intptr_t) – Describes the modes, extents and other metadata information for a tensor.

  • desc_tensor_u (intptr_t) – Describes the modes, extents and other metadata information for the output tensor U.

  • desc_tensor_v (intptr_t) – Describes the modes, extents and other metadata information for the output tensor V.

  • svd_config (intptr_t) – This data structure holds the user-requested svd parameters.

  • work_desc (intptr_t) – The workspace descriptor in which the information is collected.