
    
Ng                     L    d dl Zd dlmZ dej        fdZ G d de          ZdS )    N)OpRunc                    t          j        |           }t          |j                  }t          j        |          }|dk     r||dz   z  }|j        d|         }|j        ||         }dt          |          z  |j        z   dt          |          z  z   }	t          j        ||	          }
t          j        t          j        ||          g |d|R           }t          j        |
|k    |          S )Nr      )r   )dtype)npasarraylenshapearangereshapemod)indicesdepthaxisr   valuesrankdepth_rangelsrs	new_shapetargetss              Y/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/reference/ops/op_one_hot.py_one_hotr      s    Z  Fv|D)E""Kaxxq	af	B	d4i	 Bs2ww!22TCGG^CIji00GZvu--}}Q}}}==F:g'u5555    c                       e Zd ZddZdS )OneHotNc                 X    |\  }}t          ||||j                  }|||z
  z  |z   }|fS )N)r   r   )r   r   )selfr   r   r   r   	off_valueon_valueys           r   _runzOneHot._run   s?    $	8We$flCCCI%&2tr   )N)__name__
__module____qualname__r#    r   r   r   r      s(             r   r   )numpyr   onnx.reference.op_runr   float32r   r   r'   r   r   <module>r+      sv        ' ' ' ' ' ' #%BJ 6 6 6 6    U     r   