o
    Xic                     @  s   d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZ d dlmZmZ G dd deZd	S )
    )annotations)TypeVar)
get_schema)Opset3)BOOL	COMPLEX64
COMPLEX128DOUBLEFLOATFLOAT16INT8INT16INT32INT64STRINGUINT8UINT16UINT32UINT64)OpOpsetc                   @  sD   e Zd Zdd Zedeeeee	e
eeeeeeeeeZddd	Zd
S )Opset4c                 C  s   t | ddS )N    )r   __new__)cls r   V/home/ubuntu/.local/lib/python3.10/site-packages/onnxscript/onnx_opset/_impl/opset4.pyr   )   s   zOpset4.__new__T_Concatinputsaxisintreturnc                G  s4   t ddd}t| d|}|| j|g|R  d|iS )u"  [🌐 Concat(4)](https://onnx.ai/onnx/operators/onnx__Concat.html#concat-4 "Online Documentation")

        Concatenate a list of tensors into a single tensor

        Args:
            inputs: (variadic) List of tensors for concatenation

            axis: Which axis to concat on
        Concatr   r   r    )r   r   _prepare_inputs)selfr    r   schemaopr   r   r   r#   ?   s   zOpset4.ConcatN)r   r   r    r!   r"   r   )__name__
__module____qualname__r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r#   r   r   r   r   r   (   s*    r   N)
__future__r   typingr   	onnx.defsr   "onnxscript.onnx_opset._impl.opset3r   onnxscript.onnx_typesr   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   onnxscript.valuesr   r   r   r   r   r   r   <module>   s   D