o i@s<ddlmmZddlmmmZGdddZdS)Nc@seZdZdddZdS)numpyNcCs|dur tjjjddd}|j|_tj|j_tj|j |_ tj |j |_ tj |j |_ tj |j |_ tj|j|_tj|j|_tjj|jdd|_tj|j|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|jdd|_tjj|j dd|_ tjj|j!dd|_!tjj|j"dd|_"tjj|j#dd|_#tjj|j$dd|_$tjj|j%dd|_%tjj|j&dd|_&tjj|j'dd|_'tjj|j(dd|_(tj)|j*|_*tj)|j+|_+tj)|j,|_,tj)|j-|_-tj)|j.|_.tj)|j/|_/tj)|j0|_0tj)|j1|_1tj)|j2|_2tj)|j3|_3tj)|j4|_4tj)|j5|_5tj6|j6|_6tj7|j8|_8tj7|jj9|j_9tj7|jj9|j_9tj:|j:|_:tj;|j;|_;tj<|j<|_|_>tj|j?|_?tj|j@|_@dS) Nrnp)as_T) cumulative) num_outputs)Atracer signaturepythonimport_ndarray classicalgetitem __getitem__preserve_shapeasarray set_shapereshape transpose broadcast_toarange concatenatesplitdiagonal elementwiseaddsubtractmultiply true_divide floor_dividedivide logical_and logical_orwheremaximumminimumless less_equalgreater greater_equalequal not_equal logaddexpexplognegativedivmodreducesummeanvarstdprod count_nonzeroallanyminmaxargmaxargmintakeinplaceputatdotmatmuleinsumrollflipsortargsort)selfrrK^/home/ubuntu/.local/lib/python3.10/site-packages/einx/_src/tracer/signature/classical/numpy.py__init__srznumpy.__init__)N)__name__ __module__ __qualname__rMrKrKrKrLrsr)einx._src.tracer_srcr einx._src.tracer.signaturer rrKrKrKrLs