jettify / pytorch-optimizer

Missing module/function docstring PY-D0003
Documentation
Minor
1 occurrence in this check
Docstring missing for _matrix_power
  4from .types import OptFloat, OptLossClosure, Params
  5
  6
  7def _matrix_power(matrix: torch.Tensor, power: float) -> torch.Tensor:  8    # use CPU for svd for speed up
  9    device = matrix.device
 10    matrix = matrix.cpu()