Why it only works for matrices whose row and column are smaller than 32

[wh1994-phd]

Describe the problem Why it only works for matrices whose row and column are smaller than 32?How can i use it for higher dim?

[scaomath]

https://github.com/KinglittleQ/torch-batch-svd/blob/c0a96119187f7d55f939d2ff2b92942c6d6ca930/torch_batch_svd/batch_svd.py#L19-L21

Just change this assertion and you are good to go (of course it depends on single/double precision and the VRAM on your GPU).