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.