
    h                     P    S SK rS SKJr  S\R                  4S jr " S S\5      rg)    N)OpRunc                    [         R                  " U 5      n[        UR                  5      n[         R                  " U5      nUS:  a  X%S-   -  nUR                  SU nUR                  X% nS[        U5      -  UR                  -   S[        U5      -  -   n	[         R
                  " Xi5      n
[         R
                  " [         R                  " XA5      / UQSPUQ75      n[         R                  " X:H  US9$ )Nr      )r   )dtype)npasarraylenshapearangereshapemod)indicesdepthaxisr   valuesrankdepth_rangelsrs	new_shapetargetss              W/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/reference/ops/op_one_hot.py_one_hotr      s    ZZ Fv||D))E"Kaxq	a	B	d	 Bs2w!2!22TCG^CIjj0GZZv-}}Q}}=F::g'u55    c                       \ rS rSrSS jrSrg)OneHot   Nc                 L    Uu  pV[        XXCR                  S9nXvU-
  -  U-   nU4$ )N)r   r   )r   r   )selfr   r   r   r   	off_valueon_valueys           r   _runOneHot._run   s3    $	W$llCI%&2tr    )N)__name__
__module____qualname____firstlineno__r$   __static_attributes__r&   r   r   r   r      s    r   r   )numpyr   onnx.reference.op_runr   float32r   r   r&   r   r   <module>r/      s)     ' #%BJJ 6U r   