U ©Ê¹j<ã@sªddlmZddlZddlZddlZddlmZddlZddlm Z m Z mZmZm Z ddlmZmZmZmZdZddd d ddd ddddgZddd d ddddddœ ZGdd„dƒZdFdd„ZdddddddddddddddddddddddddddddœZdddddddddœZdd d!d"d#d$d%d&d'd(d)d*dd+œ Zd,d-„ZGd.d/„d/ƒZd0d1„ZdGd2d3d4ddd5œd6d7„ZdHdd8œd9d:„ZdIdd8œd;d<„Z d=d>„Z!d?d@„Z"dJdBdC„Z#dDdE„Z$dS)Ké)ÚannotationsN)ÚSequenceé)Ú_dtypesÚ_dtypes_implÚ_funcsÚ_ufuncsÚ_util)Ú ArrayLikeÚnormalize_array_likeÚ normalizerÚNotImplementedTypeÚC_CONTIGUOUSÚF_CONTIGUOUSÚOWNDATAÚ WRITEABLEZALIGNEDZWRITEBACKIFCOPYZFNCZFORCZBEHAVEDZCARRAYZFARRAY) ÚCÚFÚOÚWÚAÚXÚBÚCAZFAcsHeZdZddœdd„Zddœdd„Zd d „Z‡fdd„Zd d„Z‡ZS)ÚFlagsÚdict)Ú flag_to_valuecCs$tdd„| ¡Dƒƒst‚||_dS)Ncss|]}|tkVqdS©N)ÚFLAGS)Ú.0Úk©r!úT/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/_numpy/_ndarray.pyÚ 1sz!Flags.__init__..)ÚallÚkeysÚAssertionErrorÚ_flag_to_value)Úselfrr!r!r"Ú__init__0szFlags.__init__Ústr)ÚattrcCs4| ¡r | ¡tkr || ¡Std|›dƒ‚dS)NzNo flag attribute 'ú')ÚislowerÚupperrÚAttributeError)r(r+r!r!r"Ú__getattr__4szFlags.__getattr__c Csr|t ¡krt|}|tkr^z|j|WStk rZ}ztd|›ƒ|‚W5d}~XYqnXntd|›dƒ‚dS)Nzkey=ú No flag key 'r,)ÚSHORTHAND_TO_FLAGSr%rr'ÚKeyErrorÚNotImplementedError)r(ÚkeyÚer!r!r"Ú__getitem__:s$zFlags.__getitem__cs4| ¡r"| ¡tkr"||| ¡<ntƒ ||¡dSr)r-r.rÚsuperÚ__setattr__)r(r+Úvalue©Ú __class__r!r"r9EszFlags.__setattr__cCs2|tks|t ¡krtdƒ‚ntd|›dƒ‚dS)Nz"Modifying flags is not implementedr1r,)rr2r%r4r3)r(r5r:r!r!r"Ú__setitem__Ks zFlags.__setitem__) Ú__name__Ú __module__Ú__qualname__r)r0r7r9r=Ú __classcell__r!r!r;r"r/s rcs,|pˆj}‡fdd„}||_d|›|_|S)Ncs ˆ||ŽSrr!)ÚargsÚkwargs©Úfnr!r"ÚfUszcreate_method..fzndarray.)r>r@)rEÚnamerFr!rDr"Ú create_methodRs rH)ZclipZnonzeroÚrepeatÚroundZsqueezeZswapaxesÚravelZdiagonalÚdotÚtraceZargsortZsearchsortedZargmaxZargminÚanyr$ÚmaxÚminZptpÚsumÚprodZmeanÚvarZstdZcumsumZcumprodÚtakeÚchooseÚabsoluteZpositiveÚnegativeZgreaterÚlessZ greater_equalZ less_equal)ÚabsÚinvertÚposÚnegÚgtÚltÚgeÚleÚsubtractÚmultiplyÚdivideZfloor_divideÚpowerÚ remainderZbitwise_andZ bitwise_orZbitwise_xorZ left_shiftZright_shift) ÚaddÚsubÚmulÚtruedivÚfloordivÚpowÚmodÚandÚorÚxorÚlshiftÚrshiftÚmatmulcCsTt|tjƒr4|jtjtjtjtjfkrP| tj ¡Snt|t ƒrPt dd„|DƒƒS|S)Ncss|]}t|ƒVqdSr)Ú_upcast_int_indices)rÚir!r!r"r#¥sz&_upcast_int_indices..)Ú isinstanceÚtorchÚTensorÚdtypeZint8Zint16Zint32Zuint8ÚtoZint64Útuple)Úindexr!r!r"rs s rsc@seZdZdfdd„Ze ¡D]&\ZZee ep,eƒZ ee eƒeƒe<qee jdƒZee jƒZe ¡D]2\ZZee epreƒZ de›dZee eƒeƒe<q`e ¡D]z\ZZee ep®eƒZ de›dZee eƒeƒe<de›dZee fdd„eƒeƒe<d e›dZee fd d„eƒeƒe<qœee jdƒZedd„d ƒZ[[[[[ [edd„ƒZedd„ƒZedd„ƒZedd„ƒZedd„ƒZedd„ƒZedd„ƒZedd„ƒZ edd„ƒZ!ed d!„ƒZ"ed"d#„ƒZ#e#j$d$d#„ƒZ#ed%d&„ƒZ%e%j$d'd&„ƒZ%d(d)„Z&e'dgd+d,d-œd.d/„ƒZ(e'dhd+d,d-œd0d1„ƒZ)d2d3œd4d5„Z*d6d7„Z+e'd+d8œd9d:„ƒZ,d;d<„Z-d=d>„Z.ee.ƒZ/d?d@„Z0dAdB„Z1dCdD„Z2dEdF„Z3dGdH„Z4dIdJ„Z5dKdL„Z6dMdN„Z7dOdP„Z8dQdR„Z9dSdT„Z:d*dUœdVdW„Z;didYdZ„Zd_d`„Z?e j@Z@e jAZAddaœdbdc„ZBddde„ZCdS)jÚndarrayNcCs4|dkrt ¡|_nt|tjƒr(||_ntdƒ‚dS)NzSndarray constructor is not recommended; prefereither array(...) or zeros/empty(...))rvrwÚtensorruÚ ValueError)r(Útr!r!r"r)¯sÿzndarray.__init__ÚconjÚ__Ú__rcCs |||ƒSrr!©r(ÚotherrEr!r!r"ÚÍózndarray.Z__icCs||||dS)N)Úoutr!rƒr!r!r"r…Ðr†Ú __divmod__cCst ||¡Sr)rÚdivmod©r(r„r!r!r"r…Ör†Ú__rdivmod__cCst|jjƒSr)rzr}Úshape©r(r!r!r"rŒÜsz ndarray.shapecCs |j ¡Sr)r}Únumelrr!r!r"Úsizeàszndarray.sizecCs|jjSr)r}Úndimrr!r!r"räszndarray.ndimcCst |jj¡Sr)rrxr}rr!r!r"rxèsz ndarray.dtypecs&|j ¡‰t‡fdd„|j ¡DƒƒS)Nc3s|]}|ˆVqdSrr!)rÚstride©Zelsizer!r"r#ïsz"ndarray.strides..)r}Úelement_sizerzr‘rr!r’r"Ústridesìs zndarray.stridescCs |j ¡Sr)r}r“rr!r!r"Úitemsizeñszndarray.itemsizecCs(t|j ¡|jj ¡|jjdkddœƒS)NT)rrrr)rr}Ú is_contiguousÚTZ_baserr!r!r"Úflagsõs üÿz ndarray.flagscCs |j ¡Sr)r}Zdata_ptrrr!r!r"Údataszndarray.datacCs|j ¡ ¡Sr)r}ZstorageÚnbytesrr!r!r"ršszndarray.nbytescCs| ¡Sr)Ú transposerr!r!r"r— sz ndarray.TcCs t |¡Sr)rÚrealrr!r!r"rœ szndarray.realcCst|ƒj|j_dSr)Úasarrayr}rœ©r(r:r!r!r"rœscCs t |¡Sr)rÚimagrr!r!r"rŸszndarray.imagcCst|ƒj|j_dSr)rr}rŸržr!r!r"rŸscCs t |¡j}|j |¡}t|ƒSr)rrxÚtorch_dtyper}ryr|)r(rxr rr!r!r"Úastypeszndarray.astyperr r ©r(ÚordercCs| ¡Sr)Úcloner¢r!r!r"Úcopy#szndarray.copycCs t |¡Sr)rvÚflattenr¢r!r!r"r¦'szndarray.flattenF)ÚrefcheckcGsÆ|j}|rtd|›dƒ‚|dkr&dSt|ƒdkr:|d}t|tƒrJ|f}| ¡}t dd„|Dƒ¡rntdƒ‚t |¡}|| ¡kr˜|d|… |¡}n$t |¡}||d| ¡…<| |¡}||_dS) Nzresize(..., refcheck=z is not implemented.)r!rrrcss|]}|dkVqdS)rNr!)rÚxr!r!r"r#@sz!ndarray.resize..z0all elements of `new_shape` must be non-negative)r}r4ÚlenruÚintr¦ÚbuiltinsrNr~ÚmathrRrŽÚreshapervZzeros)r(r§Z new_shapeÚaZ new_numelÚretÚbr!r!r"Úresize+s* ÿ zndarray.resizecCs t |¡j}|j |¡}t|ƒSr)rrxr r}Úviewr|)r(rxr Ztviewr!r!r"r²Mszndarray.view)r:cCs|j |¡dSr)r}Zfill_ržr!r!r"ÚfillRszndarray.fillcCs |j ¡Sr)r}Útolistrr!r!r"r´Xszndarray.tolistcCst|jƒ dd¡ dd¡S)Nr}z torch.ndarrayzdtype=torch.zdtype=)r*r}Úreplacerr!r!r"Ú__str__[s ÿþÿzndarray.__str__c CsFzt ||¡WSttfk r@tj|jdtd}t|ƒYSXdS)NF)Z fill_valuerx) rÚequalÚRuntimeErrorÚ TypeErrorrvÚfullrŒÚboolr)r(r„Zfalsyr!r!r"Ú__eq__ds zndarray.__eq__cCs ||kSrr!rŠr!r!r"Ú__ne__lszndarray.__ne__cCs4zt |j ¡¡WStk r.tdƒ‚YnXdS)Nz=only integer scalar arrays can be converted to a scalar index)Úoperatorr{r}ÚitemÚ Exceptionr¹rr!r!r"Ú __index__osÿzndarray.__index__cCs t|jƒSr)r»r}rr!r!r"Ú__bool__wszndarray.__bool__cCs t|jƒSr)rªr}rr!r!r"Ú__int__zszndarray.__int__cCs t|jƒSr)Úfloatr}rr!r!r"Ú __float__}szndarray.__float__cCs t|jƒSr)Úcomplexr}rr!r!r"Ú__complex__€szndarray.__complex__cCs8z|j ¡}t|ƒ|k}Wntk r2d}YnX|S)NF)r}r¿rªrÀ)r(ÚvÚresultr!r!r"Ú is_integerƒs zndarray.is_integercCs|jjdS)Nr)r}rŒrr!r!r"Ú__len__‹szndarray.__len__cCs|j |¡Sr)r}Ú__contains__)r(r¨r!r!r"rÌŽszndarray.__contains__cGst ||¡Sr)rr›)r(Zaxesr!r!r"r›‘szndarray.transpose©r£cGstj|||dS)NrÍ)rr)r(r£rŒr!r!r"r•szndarray.reshapeéÿÿÿÿc Cst |t ||||¡¡dSr)rZcopytoÚsort)r(ZaxisÚkindr£r!r!r"rÏ™szndarray.sortcGs<|dkr|j ¡St|ƒdkr.| ¡|dS| |¡SdS)Nr!rr)r}r¿r©rKr7)r(rBr!r!r"r¿s zndarray.itemcsf|j‰‡fdd„‰t|tƒr.neg_stepc3s|]\}}ˆ||ƒVqdSrr!)rrtrÖ)r×r!r"r#¼sz&ndarray.__getitem__..r) r}rurÚtypeÚ enumerater Úndarrays_to_tensorsrsr|r7)r(r{r!)r×r}r"r7©s zndarray.__getitem__cCsFt |¡}t|ƒ}t|ƒtjkr8t|ƒ}t ||jj ¡}|j ||¡Sr)r rÚrsrØrZSCALAR_TYPESrZcast_if_neededr}rxr=)r(r{r:r!r!r"r=Ãs zndarray.__setitem__©ÚstreamcCs|jj|dS)NrÛ)r}Ú __dlpack__)r(rÜr!r!r"rÝÐszndarray.__dlpack__cCs |j ¡Sr)r}Ú__dlpack_device__rr!r!r"rÞÓszndarray.__dlpack_device__)N)r)r)rÎNN)Dr>r?r@r)ÚmethodsÚitemsÚmethodrGÚgetattrrrErHÚvarsrÚ conjugater€ÚdunderÚ ri_dunderÚplainZrvarZivarr‰rˆr‹ÚpropertyrŒrrrxr”r•r˜r™ršr—rœÚsetterrŸr¡rr¥r¦r±r²r³r´r¶Ú__repr__r¼r½rÁrÂrÃrÅrÇrÊrËrÌr›rrÏr¿r7r=rTÚputrÝrÞr!r!r!r"r|®s® ÿÿ " r|cCsNg}|D]@}t|ttfƒr"t|ƒ}t|tƒr>| |j ¡¡q| |¡q|S)z'Recusrively convert tensors into lists.)ruÚlistrzÚ_tolistr|Úappendr}r´)ÚobjZa1Úelemr!r!r"rí×s ríTÚKF)r¥r£ÚsubokÚndminÚlikec Csª|dk rtdƒ‚|dk r tdƒ‚|dkr,t‚t|tƒrT|dkrT|dkrT||jkrT|St|ttfƒrjt|ƒ}t|tƒrz|j}d}|dk r’t |¡j }t ||||¡}t|ƒS)NFz#'subok' parameter is not supported.z"'like' parameter is not supported.rñ) r4rur|rrìrzrír}rrxr r Z_coerce_to_tensor) rïrxr¥r£ròrórôr r}r!r!r"Úarrayès0ÿþýü rõ)rôcCst||||dddS)NFr)rxr£rôr¥ró)rõ)r®rxr£rôr!r!r"r srcCs(t|||d}|j ¡s$|j ¡|_|S)N)rxrô)rr}r–Ú contiguous)r®rxrôZarrr!r!r"Úascontiguousarrays r÷cCst |¡}t|ƒSr)rvÚfrom_dlpackr|)r¨rr!r!r"røs røcCs2zt |¡}Wntk r,t|ƒj}YnX|Sr)rrxrÀr)ÚentryÚdtyr!r!r"Ú_extract_dtypes rûÚsafecCs"t|ƒ}t|ƒ}t |j|j|¡Sr)rûrZ can_cast_implr )Zfrom_ryZcastingZto_r!r!r"Úcan_cast"srýcGspg}|D]R}zt|ƒj}Wn4tttfk rNt |¡}tjd|j d}YnX| |¡qtj|Ž}t |¡S)Nr)rx) rr}r¸r~r¹rrxrvÚemptyr rîrZresult_type_impl)Zarrays_and_dtypesZtensorsrùrrúr r!r!r"Úresult_type)s rÿ)N)N)Nrñ)N)rü)%Ú __future__rr«r¬r¾ÚtypingrrvÚrrrrr Z_normalizationsr rrr Znewaxisrr2rrHrßrårærsr|rírõrr÷rørûrýrÿr!r!r!r"ÚsÀõ÷ # ß%ø ó+"