torch-batch-svd icon indicating copy to clipboard operation
torch-batch-svd copied to clipboard

Published 20 hours ago • verified publisher icon KinglittleQ

SVD on complex tensors

Open jchollingsworth opened this issue 10 months ago • 0 comments

Hi, is there any possibility to add support for torch.complex64? I need to run SVD on millions of complex tensors, of size 21 by 21, and torch.svd seems a bit slow on GPU.

jchollingsworth avatar Aug 25 '23 07:08 jchollingsworth