
    Si                    \    S SK Jr  S SKrS SKJr  S\R                  4S jr " S S\5      rg)    )annotationsN)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              \/var/www/html/ai-image-ml/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   )	
__future__r   numpyr	   onnx.reference.op_runr   float32r   r   r'   r   r   <module>r1      s,    #  ' #%BJJ 6U r   