a þdYã@s6ddlmZddlZdejejeejdœdd„ZdS)é)ÚListN©é)ÚinputÚtargetÚreturncsjtt|ƒ| ¡dƒ‰| d¡‰| ˆddd¡\}}| ¡}| | dd¡ |¡¡‰‡‡‡fdd„|DƒS)aŸComputes the accuracy over the k top predictions for the specified values of k. Args: input: the input tensor with the logits to evaluate. target: the tensor containing the ground truth. topk: the expected topk ranking. Example: >>> logits = torch.tensor([[0, 1, 0]]) >>> target = torch.tensor([[1]]) >>> accuracy(logits, target) [tensor(100.)] rrTéÿÿÿÿcs6g|].}ˆdt|ˆƒ… d¡ ¡ d¡dˆ‘qS)NrrgY@)ÚminÚreshapeÚfloatÚsum)Ú.0Úk©Z batch_sizeÚcorrectZmaxk©ú`/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/metrics/accuracy.pyÚ ózaccuracy..)r ÚmaxÚsizeÚtopkÚtÚeqr Z expand_as)rrrÚ_ÚpredrrrÚaccuracys  r)r)ÚtypingrZtorchZTensorrrrrrÚs