Is there any function in pytorch to compute the top_k singular values / vectors of a matrix ?
For now I’m using the torch.svd
function, but it’s total overkill in some situations where I only need the first singular value/vectors.
Note that this is to implement an Optimizer, therefore I don’t need gradients.