
    bIg                    f    d dl mZ d dlZd dlmZ  G d de      Z G d de      Z G d d	e      Zy)
    )annotationsN)OpRunc                      e Zd ZddZy)Unsqueeze_1Nc                   t        |t        j                        rt        |      }n$|g dfv rd }nt        |t              rt        |      }t        |t        t        f      r#|}|D ]  }t        j
                  ||      } |fS t        d      )N axisz9axes cannot be None for operator Unsqueeze (Unsqueeze_1).)
isinstancenpndarraytuplelistexpand_dimsRuntimeErrorselfdataaxessqas        \/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/reference/ops/op_unsqueeze.py_runzUnsqueeze_1._run   s    dBJJ';Db"XDd#;DdUDM*B 0^^BQ/0 u K     N__name__
__module____qualname__r   r   r   r   r   r      s    r   r   c                      e Zd Zy)Unsqueeze_11N)r   r   r   r   r   r   r!   r!      s    r   r!   c                      e Zd ZddZy)Unsqueeze_13Nc                   |bt        |d      r<t        |j                        dkD  r$	 t        j                  |t        |            }|fS t        j                  ||      }|fS t        d      # t        $ r` t        |      dk(  r$t        j                  |t        |      d         }n)|}t        |      D ]  }t        j                  ||      } Y |fS w xY w)N__iter__r   r	      z:axes cannot be None for operator Unsqueeze (Unsqueeze_13).)	hasattrlenshaper   r   r   	TypeErrorreversedr   r   s        r   r   zUnsqueeze_13._run#   s    tZ(S_q-@	<5;?B u ^^Dt4
 u L  ! <4yA~^^DuT{1~F!!)$ <A!#!;B< u<s    A0 0A$CCr   r   r   r   r   r#   r#   "   s    r   r#   )	
__future__r   numpyr   onnx.reference.op_runr   r   r!   r#   r   r   r   <module>r/      s4    #  '% &	; 	5 r   