cuquantum.cutensornet.tensor_svd

cuquantum.cutensornet.tensor_svd(intptr_t handle, intptr_t tensor_in, intptr_t raw_data_in, intptr_t tensor_u, intptr_t u, intptr_t s, intptr_t tensor_v, intptr_t v, intptr_t config, intptr_t info, intptr_t workspace, intptr_t stream)[source]

Perform SVD decomposition of a tensor.

Parameters

  • handle (intptr_t) – The library handle.

  • tensor_in (intptr_t) – The input tensor descriptor.

  • raw_data_in (intptr_t) – The pointer address (as Python int) to the input tensor (on device).

  • tensor_u (intptr_t) – The tensor descriptor for the output U.

  • u (intptr_t) – The pointer address (as Python int) to the output tensor U (on device).

  • s (intptr_t) – The pointer address (as Python int) to the output array S (on device).

  • tensor_v (intptr_t) – The tensor descriptor for the output V.

  • v (intptr_t) – The pointer address (as Python int) to the output tensor V (on device).

  • config (intptr_t) – The tensor SVD config handle.

  • info (intptr_t) – The tensor SVD info handle.

  • workspace (intptr_t) – The workspace descriptor.

  • stream (intptr_t) – The CUDA stream handle (cudaStream_t as Python int).

Note

After this function call, the output tensor descriptors tensor_u and tensor_v may have their shapes and strides changed. See the documentation for further information.

See also

cutensornetTensorSVD()