o €o™ieã@s*ddlmZddlmZGdd„dƒZdS)é)ÚUnion)ÚTensorc@s”eZdZUdZeeeefed<eeefed<eeefed<e ed<ddd „Z dd d „Z ddeeeefd e ddfdd„Z e defdd„ƒZdS)Ú AverageMetera"Computes and stores the average and current value. Example: >>> stats = AverageMeter() >>> acc1 = torch.tensor(0.99) # coming from K.metrics.accuracy >>> stats.update(acc1, n=1) # where n is batch size usually >>> round(stats.avg, 2) 0.99 ÚvalÚ_avgÚsumÚcountÚreturnNcCs | ¡dS©N)Úreset©Úself©rúP/home/ubuntu/.local/lib/python3.10/site-packages/kornia/metrics/average_meter.pyÚ__init__(s zAverageMeter.__init__cCsd|_d|_d|_d|_dS)Nr)rrrrr rrrr +s zAverageMeter.resetéÚncCs8||_|j||7_|j|7_|j|j|_dSr )rrrr)r rrrrrÚupdate1szAverageMeter.updatecCs t|jtƒr t|j ¡ƒS|jSr )Ú isinstancerrÚfloatÚitemr rrrÚavg7s zAverageMeter.avg)r N)r)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrÚboolrÚ__annotations__Úintrr rÚpropertyrrrrrrs    "rN)ÚtypingrÚ kornia.corerrrrrrÚs