o y“©i¥ã @sèddlmZddlZddlmZddlmZdededefdd „Zdededefd d „Zd ed edefdd„Zdededeeeffdd„Z ddedede de def dd„Z ddedede defdd„Zddedede defdd„ZdS)é)ÚTupleN)ÚTensor)Ú_TORCH_GREATER_EQUAL_1_9ÚxÚyÚreturncCs8|jtjks |jtjkr| ¡|j ¡ ¡S||jS)z~Safe calculation of matrix multiplication. If input is float16, will cast to float32 for computation and back again. )ÚdtypeÚtorchÚfloat16ÚfloatÚTÚhalf©rr©rúR/home/ubuntu/.local/lib/python3.10/site-packages/torchmetrics/utilities/compute.pyÚ _safe_matmuls rcCs|t |¡}d||dk<|S)z§Computes x * log(y). Returns 0 if x=0. Example: >>> import torch >>> x = torch.zeros(1) >>> _safe_xlogy(x, 1/x) tensor([0.]) çr)r Úlog)rrÚresrrrÚ _safe_xlogy s rÚnumÚdenomcCs<d||dk<| ¡r |n| ¡}| ¡r|n| ¡}||S)zSafe division, by preventing division by zero. Additionally casts to float if input is not already to secure backwards compatibility. ér)Úis_floating_pointr )rrrrrÚ _safe_divide/s rcCsŒ|jdkr | ¡n|}|jdkr| ¡n|}|jdks |jdkr,td|j›d|j›ƒ‚| ¡| ¡krBtd| ¡›d| ¡›ƒ‚||fS)z!Checks that auc input is correct.rzJExpected both `x` and `y` tensor to be 1d, but got tensors with dimension z and zHExpected the same number of elements in `x` and `y` tensor but received )ÚndimÚsqueezeÚ ValueErrorÚnumelrrrrÚ_auc_format_inputs:sÿÿréÿÿÿÿÚ directionÚaxiscCsBt ¡tj|||d|}Wdƒ|S1swY|S)zrComputes area under the curve using the trapezoidal rule. Assumes increasing or decreasing order of `x`. )ÚdimN)r Úno_gradÚtrapz)rrr!r"Úauc_rrrÚ_auc_compute_without_checkJs  ÿþr'FÚreordercCs¦t ¡E|rtrtj|ddnt |¡\}}||}|dd…|dd…}|dk ¡r:|dk ¡r6d}ntdƒ‚d}t|||ƒWdƒS1sLwYdS) NT)Ústablerr rgð¿z_The `x` tensor is neither increasing or decreasing. Try setting the reorder argument to `True`.gð?)r r$rÚsortÚanyÚallrr')rrr(Úx_idxÚdxr!rrrÚ _auc_computeTs    ÿ $ñr/cCst||ƒ\}}t|||dS)a8Computes Area Under the Curve (AUC) using the trapezoidal rule. Args: x: x-coordinates, must be either increasing or decreasing y: y-coordinates reorder: if True, will reorder the arrays to make it either increasing or decreasing Return: Tensor containing AUC score )r()rr/)rrr(rrrÚaucgs r0)r )F)Útypingrr rÚtorchmetrics.utilities.importsrrrrrr Úintr'Úboolr/r0rrrrÚs