4X z2 d dl Z d dlZd dlZd dlmZmZmZmZm Z m
Z
m Z m Z dZ
ej g d Z e ee
e Z G d d Z G d d Zd ZdS )
N)assert_ assert_equalassert_array_equalassert_almost_equal
assert_raisessuppress_warningsassert_raises_regexassert_allclose
abcdefghij) r r
r r r r
c e Zd Zd Zd Zd Z ej d1d Zd Z d Z
d Z d Z d
Z
d Zd Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z e!j" # d ej$ d ej$ d z d! Z%d" Z&d# Z'd2d%Z(d& Z)d' Z*d( Z+d) Z,d* Z-d+ Z.d, Z/d- Z0d. Z1d/ Z2d0 Z3d$S )3
TestEinsumc x dD ]}t t t j | t t t j d| t t t j dd| t t t j ddd| t t t j ddd| t t t j ddd |
t t t j ddd | t t t j ddd|
t t t j gdR d|i t t t j ddd| t t t j dddgdg| t t t j ddg| t t t j dd| t t t j dddg| t t t j dd| t t t j dddg| t t t j dd| t t t j dddg| t t t j dddg| t t t j dddg| t t t j dddg| t t t j dddg| t t t j dddg| t t t j d ddg| t t t j d ddg| t t t j d ddg| t t t j d ddgddgg| t t t j d t j d! d"d# | t t t j d$t j d! d"d# | t t t j dt j d! d"d# | t t t j d%dd&gdd&ggt j d' d"d" | t t d( 5 t j d) }t j d* }t j d+|| d d d n # 1 swxY w Y t t t j d%t j d! d,d& |d-. d S )/NTFoptimize r test)outr Worderr blah)castingr
bad_data_type)dtyper )bad_argr )?NNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNr ,iijz...izi...jzi...zij...zi..z.i...zj->..jzj->.j...zi%...z...j$zi->&zi->ijzij->jijiir r r
ii->izi->i r z'b')r
r
r r r )r
r r z aabcb,abcd)r r ) r
ValueErrornpeinsum TypeErrorarangereshaper ones)selfdo_optabs p/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/core/tests/test_einsum.pytest_einsum_errorsz TestEinsum.test_einsum_errors s # R 6 R 6F *bi& A A A A *bif E E E E
)RY1v F F F F
)RYA6#)
+
+
+
+
*biQc#)
+
+
+
+
*biQ#)
+
+
+
+
)RYA_#)
+
+
+
+
)RYAq#)
+
+
+
+
)RY M M M Mf M M M
*biQF K K K K *bia!qc#)
+
+
+
+ *biqcF K K K K
*bia& I I I I *bi1v O O O O *biV L L L L *bi1a&6 R R R R *biV L L L L *bi1a&6 R R R R
*biA P P P P *bi1a&6 R R R R *biAq6F S S S S *biaVf U U U U
*bi1a&6 R R R R *bi1a&6 R R R R *bi!Q& Q Q Q Q
*bi1a&6 R R R R
*biaVaV.CustomExceptionN)__name__
__module__ __qualname__ r5 CustomExceptionr9 k s Dr> r? c 6 e Zd Zd Z fdZd Z fdZd ZdS ):TestEinsum.test_einsum_object_errors..DestructoBoxc " || _ || _ d S N_val _destruct)r1 valuedestructs r5 __init__zCTestEinsum.test_einsum_object_errors..DestructoBox.__init__p s ! !)r> c N | j |j z }|| j k r|| _ | S rC rD r1 othertmpr? s r5 __add__zBTestEinsum.test_einsum_object_errors..DestructoBox.__add__t 0 i%*,$.(()) #DI Kr> c < |dk r| S | | S Nr )rN r1 rL s r5 __radd__zCTestEinsum.test_einsum_object_errors..DestructoBox.__radd__| " A:: K <<...r> c N | j |j z }|| j k r|| _ | S rC rD rK s r5 __mul__zBTestEinsum.test_einsum_object_errors..DestructoBox.__mul__ rO r> c < |dk r| S | | S rQ )rV rR s r5 __rmul__zCTestEinsum.test_einsum_object_errors..DestructoBox.__rmul__ rT r> N)r: r; r< rI rN rS rV rX )r? s r5 DestructoBoxrA n st
*
*
*
/
/
/
/
/
/
/
/r> rY c ( g | ]} |d S )r r= .0r# rY s r5
z8TestEinsum.test_einsum_object_errors.. s% ???Qll1a((???r> r'
objectr r
zij->ic ( g | ]} |d S )d r= r[ s r5 r] z8TestEinsum.test_einsum_object_errors.. s% AAAqll1c**AAAr> r z
i...k->...c ( g | ]} |d S )7 r= r[ s r5 r] z8TestEinsum.test_einsum_object_errors.. s% ???all1b))???r> r ij, jzij, jhzij->) Exceptionr+ arrayranger/ r r, )r1 r3 r4 r? rY s @@r5 test_einsum_object_errorsz$TestEinsum.test_einsum_object_errorsg s i / / / / / / / / / /D
H????%2,,???#
%
%
%%,WQ]]
ory'1===
HAAAAE!RLLAAA#
%
%
%%,WQ1%5%5
ory,BBB
H????5A;;???#
%
%
%ory'1a@@@ ory(AqAAA ory&!<<<< c
, dD ]}t j d d_ t j d| }t |j u t j t g| }t |j u t j d| }t |j u t | t j ddg| }t |j u t | t j d| }t |j d d
j d < t j d| }t |j d t j d d_ t j d | }t |j u t |j t j ddg| }t |j u t |j t j d d
_ t j d| }t |j u t |fdt d D t j ddgdg| }t |j u t |fdt d D t j d d_ t j d| }t |j u t |d D t j t ddgt dg| }t |j u t |d D t j d| }t |j u t |d ddd D t j ddt gt dg| }t |j u t |d ddd D t j d| }t |j u t |fd t d D t j t ddgdt g| }t |j u t |fd t d D t j d | }t |j u t |fd t d D t j g d ddg| }t |j u t |fd!t d D t j d"| }t |j u t |fd#t d D t j ddt gdt g| }t |j u t |fd$t d D t j d%| }t |j u t |fd&t d D t j dt dgdt g| }t |j u t |fd't d D t j d(| }t |j u t |d) ddd D t j dt dgt dg| }t |j u t |d* ddd D t j d d_ t j d+| }t |j u t |fd,t d D t j g d-dg| }t |j u t |fd.t d D t j d/ d0_ t j d1| }t |j u t | dd t j g d2g d3| }t |j u t | dd d S )4Nr r r r
z...r r$ r r' WRITEABLEFji )r
r
r& c $ g | ] }||f
S r= r= r\ r# r3 s r5 r] z0TestEinsum.test_einsum_views.. ! 7 7 7Qq!tW 7 7 7r> r
c $ g | ] }||f
S r= r= rq s r5 r] z0TestEinsum.test_einsum_views.. rr r> rc )r
r
r
z ...ii->...ic F g | ] fd t d D S )c $ g | ] }||f
S r= r= r\ r# xs r5 r] z;TestEinsum.test_einsum_views... ! 8 8 8!a1g 8 8 8r> r
ri r\ rw s @r5 r] z0TestEinsum.test_einsum_views.. 4 D D DQ 8 8 8 8uQxx 8 8 8 D D Dr> c F g | ] fd t d D S )c $ g | ] }||f
S r= r= rv s r5 r] z;TestEinsum.test_einsum_views... rx r> r
ry rz s @r5 r] z0TestEinsum.test_einsum_views.. r{ r> z ii...->...ic F g | ] fd t d D S )c $ g | ] }||f
S r= r= rv s r5 r] z;TestEinsum.test_einsum_views... rx r> r
ry rz s @r5 r] z0TestEinsum.test_einsum_views.. C < < r c F g | ] fd t d D S )c $ g | ] }||f
S r= r= rv s r5 r] z;TestEinsum.test_einsum_views... rx r> r
ry rz s @r5 r] z0TestEinsum.test_einsum_views.. r r> z ...ii->i...c * g | ]}d d ||f S rC r= rq s r5 r] z0TestEinsum.test_einsum_views.. ' : : :AQqqq!QwZ : : :r> c * g | ]}d d ||f S rC r= rq s r5 r] z0TestEinsum.test_einsum_views.. r r> zjii->ijc * g | ]}d d ||f S rC r= rq s r5 r] z0TestEinsum.test_einsum_views.. r r> )r' r r c * g | ]}d d ||f S rC r= rq s r5 r] z0TestEinsum.test_einsum_views.. r r> z ii...->i...c T g | ]$} d dd dd||f %S r r r' N transposerq s r5 r] z0TestEinsum.test_einsum_views.. 8 M M MqQ[[Aq 1 1!!!Q' : M M Mr> c T g | ]$} d dd dd||f %S r r rq s r5 r] z0TestEinsum.test_einsum_views.. r r> z i...i->i...c T g | ]$} d dd dd||f %S r' r r Nr rq s r5 r] z0TestEinsum.test_einsum_views.. r r> c T g | ]$} d dd dd||f %S r r rq s r5 r] z0TestEinsum.test_einsum_views..
r r> z i...i->...ic F g | ] fd t d D S )c $ g | ] }||f
S r= r= rv s r5 r] z;TestEinsum.test_einsum_views... rx r> r
ry rz s @r5 r] z0TestEinsum.test_einsum_views.. r r> c F g | ] fd t d D S )c $ g | ] }||f
S r= r= rv s r5 r] z;TestEinsum.test_einsum_views... rx r> r
ry rz s @r5 r] z0TestEinsum.test_einsum_views.. r r> ziii->ic & g | ]
}|||f S r= r= rq s r5 r] z0TestEinsum.test_einsum_views.. # : : :AQq!QwZ : : :r> )r r r c & g | ]
}|||f S r= r= rq s r5 r] z0TestEinsum.test_einsum_views..# r r> r r
r zijk->jikr r' r )r' r r )
r+ r. shaper, r baseEllipsisr flagsTri r swapaxes)r1 r2 r4 r3 s @r5 test_einsum_viewsz TestEinsum.test_einsum_views s # G . G .F ! A AG %V444A AFaK !hZ&999A AFaK $F333A AFaK A !aVf555A AFaK A %V444A AGK( ) ) )#(AGK %V444A ,, - - - ! A AG $F333A AFaK AC !aVf555A AFaK AC ! A AG '1v666A AFaK 7 7 7 7eAhh 7 7 7 8 8 8 !aVaS6:::A AFaK 7 7 7 7eAhh 7 7 7 8 8 8 "
A AG -V<< Fc
: t j | }t dd D ]}t j || }t j |d }t |d r| | }t t j d|| | t t j |d gg | | t dd D ]}t j d
|z | d d | }t j |d }t |d r| | }t t j d
|| | t t j |t d gt g| | t dd D ]}t j d |z | d | }t j |d }t |d r| | }t t j d|| | t t j |d t gt g| | t dd D ]}t j d
|z | d d | }t j |d }t |d r| | }t t j d|| | t t j |d t gt g| | t dd D ]"}t j ||z | || }t j | }t |d r| | }t t j d|| | t t j |d d g| | t j d d g }t t j ||| | t t j |t | | | $t t j dd d d t dd D ]}t j d |z | d | }t j d
|z | d d | }t t j d||| t j || t t j |t g|t g| t j || t dd D ]}t j d
|z | d d | }t j || }t t j d||| t j || t t j |t d g|t d g| t j || t dd D ]}t j |d z d z | |d d }t j || }t t j d||| t j |j |j j t t j |d t g|d t g| t j |j |j j t dd D ]}t j d | dz }t j || dz }t t j d||| t j || t t j |d g|dg| t j || t% 5 }| t j t dd D ]}t j d|z | d| }t j || }t t j d||| t j || t t j |d dg|dg| t j || t j d| }t j d|||dd| t |t j | d | d | d |d < t j |d dg|dg|dd| t |t j | d | d | t dd D ]}t j d|z | d| }t j || }t t j d |j |j | t j |j |j t t j |j dd g|j dg| t j |j |j t j d| }t j d |j |j |dd| t |t j |j d |j d | d |d < t j |j dd g|j dg|dd| t |t j |j d |j d | t dd D ]}|d k s|d k rt j d|z | d| }t j |d
z | |d
}t t j d ||| t j || t t j |d dg|dd g| t j || t dd D ]}t j d|z | d| }t j |d
z | |d
}t j d!| dd
}t j d |||dd| t |t j | d | d | d |d < t j |d dg|dd g|dd| t |t j | d | d | t j d| d d }t j d"| dd# }t j d$| d#d
}|d k rt t j d%|||| | | | t t j |d dg|dd g|d d g| | | | t j d&| d d
} t j d%|||| dd| | d | d }
|
| d | }
t | |
d | d < t j |d dg|dd g|d d g| dd|
| d | d }
|
| d | }
t | |
t j | t j d k rt j d'| d dd# }t j d!| dd d }t t j d(|| t j ||dd gd dgf) t t j |g d*|g d+d d g t j ||dd gd dgf) t j d,| d#d }t j d-|||dd| t |t j | d | d dd gd dgf) | d |d < t j |g d*|g d+d d g|dd| t |t j | d | d dd gd dgf) | d d d n # 1 swxY w Y |j d.k rd/nt j | j dz
} t j dd | d dd0d dg| }t j d d1d2| d dd dg| }t j g d3 }t t j d4|||d5d|6 t j t j |d k |d k |d k t t j |d g|d g|d gd gd5d7 t j t j |d k |d k |d k t j d8| }t t j d9d | d t j | z t t j d g |d gg d t j | z t t j d:|d d t j | z t t j |d gd g g d t j | z t dd; D ]+}t j || }t j | j dk rt t j d<||| t j || t t j d=||| t j || t t j d>|d | d |z t t j d?d || d |z t t j d:|d | d t j | z t t j d9d || d t j | z t t j d<|dd |d d | t j |dd |d d t t j d=|dd |d d | t j |dd |d d t t j d>|dd d | d |dd z t t j d?d |dd | d |dd z t t j d:|dd d | d t j |dd z t t j d9d |dd | d t j |dd z -t j d8t: }t j d||d7 }t |t j | t |d@ r't |j t j | t j |d gg |d7 }t |t j | t |d@ r't |j t j | t j d dz } t j d d d d z }
t j d d d dAz }t t j dB| |
| dC t j dD } t j dE }
t? t j dF| |
dG t j dF| |
dH t? t j dF| |
dG dIgd z t j dJdKg }t j dLg }t? t j dM||dH dN t? t j dM||dG dN t j dO d z } t j dP d z }
dQD ]w}t? t j dR| | | t j dR| |
| t? t j dR| |
| t j dOdS xt j! d | }t j d | }t? t j dT||| dJg t? t j dU||| dJg t? t j dV||| dJg d S )WNr' r` r( )axisastypei->r r r r r
z ...i->...z i...->...r% z..., ...r z
...i, ...i z
i..., i...i,jrf zij,jf8unsafe)r r r r .zji,j f2zij,jkr r zij,jk,kl < zijk, jil -> klaxesr )r' r r
r^ z ijk,jil->klu
g @g )TTFTTFTTzi,i,i->i?)r r r )r r ro ,i->i,-> z...,...zi,ii,->i,i->ir z z,mz,zm-> )r^ r r' r zij,ij->jTFg $@g @ @g @zi, ig 4@)r' r r r r ...ij,...jk->...ikg ?zji,i->zi,ij->zij,i->)"r+ r ri r. sumhasattrr r r, r/ r traceasarraylistmultiplyinnerr outerr filterComplexWarningdot tensordotkindiinfomaxrh logical_anditemsizer_ r0 r fulleye)r1 r r2 nr3 r4 np_arraysupcr) tgtneg_valpqrrw yr s r5 check_einsum_sumsz TestEinsum.check_einsum_sums1 s: q" D DA !5)))Aqr"""Aq(##
$HHUOO 5!f===q A A A 1qc2??? C C C Cq" V VA #a%u---55aA>>Aqr"""Aq(##
$HHUOO ;FCCCQ G G G 1xmhZ&QQQST U U U U q" V VA !A#U+++33Aq99Aqq!!!Aq(##
$HHUOO ;FCCCQ G G G 1q(mhZ&QQQST U U U Uq" V VA #a%u---55aA>>Aqq!!!Aq(##
$HHUOO ;FCCCQ G G G 1q(mhZ&QQQST U U U U q" K KA !A#U+++33Aq99A Aq(##
$HHUOO 4V<<>> B B B z1a&))H 1h@@@! D D D 1d8nnvFFF J J J J RYz1a00"555q" , ,A !a%u---55a;;A %!)511199!QBBA :q!fEEEQ**
,
,
, 1xj!hZ&QQQQ**
,
,
,
, q" ) )A %!)511199!QBBA !5)))A <AGGGRSUV X X X 1xmQ1
PVWWW!Q
)
)
)
) q" / /A !a%!)511199!QBBA !5)))A <AGGG!#qs++-
/
/
/ 1q(mQH
PVWWW!#qs++-
/
/
/
/ q" ) )A !5)))!+A !5)))!+A 5!Q@@@!Q
)
)
) 1qc1qcFCCC!Q
)
)
)
)
s CC JJr( ) ) ) 1b\\
C
CIac///771==Iau--- RYw1v F F F VAq\\+ + + RYq1a&!aS6 J J J VAq\\+ + + Iau--- &!QA $hI I I I Q VAHHTNN$%HHTNN 4 44:F5MMC C C # !aVQ $hI I I I Q VAHHTNN$%HHTNN 4 44:F5MMC C C C 1b\\
E
EIac///771==Iau--- RYvqsAC& I I I VAC - -/ / / RYqsQFAC!v N N N VAC - -/ / / Iau--- &!#qs $hI I I I Q VACJJt$4$4$%CJJt$4$4 6 66 > \!Qq!fq!f5E F F FH H H RYq)))Q Aq6 J J \!Qq!fq!f5E F F FH H H Ib...66q!<< -A1 $hI I I I Q QXXd^^QXXd^^$%q6Aq6"2!4 !4 !44:F5MMC C C# !YYY999q!f! $hI I I I Q QXXd^^QXXd^^$%q6Aq6"2!4 !4 !44:F5MMC C Ces C s C s C s C s C s C s C s C s C s C s C s C s C s C s Cl
c))""rx/BQ/F Ha1R"Q?u M M M Ha3Ga1bA O O O HGGG H HRYz1a 6C C C^BN1616$B$BAFKK M M M RYq1#q1#q1#s 2 2 2^BN1616$B$BAFKK M M M
Iau % % %RYvq!,,aq k:::RYq"a!b111RVAYY;???RYvq!,,aq k:::RYq1#q"b111RVAYY;??? q" . .A !5)))Ax'!++ RYy!Q H H H [A . .0 0 0 RYuaV D D DbfQPQllSSS RYw1v F F F!LLL RYw1v F F F!LLL RYvq!f E E EqPQ{SSS RYvq!f E E EqPQ{SSS RYy!ABB%3B3& Q Q Q [122#2# 7 79 9 9 RYuaeQssVf M M M VAabbE1SbS6 2 24 4 4 RYw!""q6 J J JAaPQPRPReGTTT RYw1QRR56 J J JAaPQPRPReGTTT RYvqua& I I I rvae}}_. . . RYvq!ABB%& I I I rvae}}_. . .
Iav & & & IeQeX > > >Qq """ 1g 3 "(5// 2 2 2 Ia!bx @ @ @Qq """ 1g 3 "(5// 2 2 2
IaLL1 IaLL A & & * IaLL A & & *RY{Aq!44c:::
GFOO GENN29ZAEEE 9ZAFFF H H H29ZAEEE 519 & & &
Hb"X HbTNN29VQEBBBCHHH29VQDAAA3GGG
GFOOa GFOOa % 6 6H ry)=q!2: < < < !y)=q!2: < < <
=
=
=
ry)=q!2: < < < !wvt 4 4
6
6
6
6
F1E " " " GAU # # #29Xq!hGGG 4 ! ! !29Xq!hGGG 4 ! ! !29Xq!hGGG 4 ! ! ! ! !s nAN.N.AN2N5AN2c 0 | d d S )Ni1r r1 s r5 test_einsum_sums_int8z TestEinsum.test_einsum_sums_int8c t$$$$$r> c 0 | d d S )Nu1r r s r5 test_einsum_sums_uint8z!TestEinsum.test_einsum_sums_uint8f r r> c 0 | d d S )Ni2r r s r5 test_einsum_sums_int16z!TestEinsum.test_einsum_sums_int16i r r> c 0 | d d S )Nu2r r s r5 test_einsum_sums_uint16z"TestEinsum.test_einsum_sums_uint16l r r> c \ | d | dd d S )Ni4Tr r s r5 test_einsum_sums_int32z!TestEinsum.test_einsum_sums_int32o 2 t$$$ tT*****r> c \ | d | dd d S )Nu4Tr r s r5 test_einsum_sums_uint32z"TestEinsum.test_einsum_sums_uint32s r r> c 0 | d d S )Ni8r r s r5 test_einsum_sums_int64z!TestEinsum.test_einsum_sums_int64w r r> c 0 | d d S )Nu8r r s r5 test_einsum_sums_uint64z"TestEinsum.test_einsum_sums_uint64z r r> c 0 | d d S )Nr r r s r5 test_einsum_sums_float16z#TestEinsum.test_einsum_sums_float16} r r> c 0 | d d S )Nf4r r s r5 test_einsum_sums_float32z#TestEinsum.test_einsum_sums_float32 r r> c \ | d | dd d S )Nr Tr r s r5 test_einsum_sums_float64z#TestEinsum.test_einsum_sums_float64 r r> c D | t j d S rC )r r+
longdoubler s r5 test_einsum_sums_longdoublez&TestEinsum.test_einsum_sums_longdouble s r}-----r> c \ | d | dd d S )Nc8Tr r s r5 test_einsum_sums_cfloat64z$TestEinsum.test_einsum_sums_cfloat64 r r> c 0 | d d S )Nc16r r s r5 test_einsum_sums_cfloat128z%TestEinsum.test_einsum_sums_cfloat128 s u%%%%%r> c D | t j d S rC )r r+ clongdoubler s r5 test_einsum_sums_clongdoublez'TestEinsum.test_einsum_sums_clongdouble s r~.....r> c \ | d | dd d S )Nr_ Tr r s r5 test_einsum_sums_objectz"TestEinsum.test_einsum_sums_object s2 x((( x.....r> c ~ t j d }t j d }t t j d|| dgdggg t t j d||d dgdggg t t j d|| dgdggg t t j dg dg d d
t t j dg dg d d d
t j d t j }t j d
t j }t t j d||| t j d|| t t j d|||d t j d||d t j dd }t j dd dd }t j dd dd }t t j d||| ddgddgddgddggddgddgd d gdd ggg t t j d|||d ddgddgddgddggddgddgd d gdd ggg t t j d dgdgd dgg d S )!Nr )r r r' zij...,j...->i...r Tr z ...i,...i)r' r r
r r greedy)r r r r r
)r r r zijklm,ijn,ijn->z ijklm,ijn->r' r
r ro r z x,yx,zx->xzy r r r 0 @ r r )r+ r0 r r, int64r. r/ )r1 r3 r4 r s r5 test_einsum_misczTestEinsum.test_einsum_misc s
GFOO GI RY 11a88QC!:,GGGRY 11a$GGGA3QRPS* VVV RY 11a88QC!:,GGGRY{IIIyyyAA2FFFRY{IIIyyy(02 2 235 7 7 7
G$bh / / GK * *RY 0!Q::Y}a33 5 5 5RY 0!QDIIIY}aTBBB D D D
IaOO IaOO # #Aq ) ) IaOO # #Aq ) )RY~q!Q77AwQ!R1b':"gBx"bB8<> ? ? ? RY~q!QFFFAwQ!R1b':"gBx"bB8<> ? ? ? RYuqcA3D999QC5AAAAAr> c G d d }t j | g }t j dt j d j z dt
}t
j t 5 t j ddg| d d d n # 1 swxY w Y t
j t 5 t j d|dg d d d n # 1 swxY w Y t j d|| dk sJ d S ) Nc e Zd Zd ZdS ))TestEinsum.test_object_loop..Multc dS )N* r= rR s r5 rV z1TestEinsum.test_object_loop..Mult.__mul__ s rr> N)r: r; r< rV r= r> r5 Multr s#
r> r" r r' )bufferr r r r! )
r+ rh ndarrayintpr r_ pytestraisesr- r, )r1 r" objMultobjNULLs r5 test_object_loopzTestEinsum.test_object_loop s (DDFF8$$*ebgajj.A&ARXYYY
]9
%
% + + IeaS' * * * + + + + + + + + + + + + + + +
]9
%
% + + IeWqc * * * + + + + + + + + + + + + + + +y11R777777s$ :B B"%B"C&&C*-C*c ~ t j d t j d t j ddgddgddgd t j ddgddgddgd t j dd gd dgddgd t t fd
t t fd d S ) Nrl r
r r r r Fr rc 3 c D t j ddgddgddgd S )Nr 4 r Fr r+ r, r3 r4 s r5 z1TestEinsum.test_subscript_range.. s, ")A2wB7QPQF]b*c*c*c r> c D t j ddgddgddgd S )Nr( r r Fr r1 r2 s r5 r3 z1TestEinsum.test_subscript_range.. s, ")AAwAq6BPQ7]b*c*c*c r> )r+ r0 r, r r* r1 r3 r4 s @@r5 test_subscript_rangez TestEinsum.test_subscript_range s
GFOO GFOO
!aWa"a1a&5AAAA
!aWa"a1a&5AAAA
!aWa"a1a&5AAAAj"c"c"c"c"cdddj"c"c"c"c"cdddddr> c t j d ddd }t j d }t j d||d }dD ]t}t t j d ||| | t t j d
||| | t t j d ||| | ut j d d
}t j d d }t j d||d }dD ]}t t j d||| | t t j d||| | t t j d||| | t t j d||| | g d}t j t j | | }t j |d }t j d||d }dD ]t}t t j d||| | t t j d||| | t t j d||| | ud\ }} }
t j || z |
z ddd|| |
}t j || z |
z dz || |
d }t j d ||d }dD ](}t t j d ||| | )d S ) Nr r r
r z
ijk,j->ijkFr r zij...,j...->ij...zij...,...j->ij...zij...,j->ij...r r r
r )r
r z ik,kj->ijzik...,k...->i...zik...,...kj->i...jz...k,kjz
ik,k...->i...)r r
r r z ijkl,k->ijlzijkl,kz...kl,kz
...kl,k...) r9 x r' z...lmn,...lmno->...oz...lmn,lmno->...o)r+ r. r/ r, r prod) r1 ABrefoptdimsr3 vJKMs r5 test_einsum_broadcastz TestEinsum.test_einsum_broadcast sb
Ii ( (Aq 1 1 IaLLi aU;;; O OC #61sKKKS Q Q Q #61sKKKS Q Q Q #3QCHHH# N N N N IbMM ! !& ) ) IaLL ( (i QE::: N NC #5q!cJJJC P P P #7ALLLc R R R 9aSAAA3 G G G ?Aq3GGG M M M M|| Ibgdmm $ $ , ,T 2 2 Id1g i
q!e<<< K KC 8QC@@@# F F F 9aSAAA3 G G G <ADDDc J J J J 1a Ia!eai ( (Aq!Q : : Ia!eai!m $ $ , ,Q1a 8 8i.1uEEE 7 7C #61,/1 1 125
7
7
7
7 7 7r> c ( t j d dd t j }t j d ddd t j }t j d|| }t j ||d }t || t j d dd t j }t j d
ddd d t j }t j d || }t j ||d }t || d S )
Nr r r
i@ i
z cl, cpx->lpx)r r r ro i r zcl, cpxy->lpxy)
r+ r. r/ r float32int16r, r r float64)r1 r< r= estps r5 test_einsum_fixedstridebugz%TestEinsum.test_einsum_fixedstridebug sA "
Ie $ $Q * * 1 1"* = = Il # # + +Aq$ 7 7 > >rx H H
Y~A
.
.
\!aw
/
/
/R"
Ie $ $Q * * 1 1"* = = Io & & . .q!R < < C CBJ O O
Y'!Q
/
/
\!Qf
-
-
-R r> c & t j ddd }t j d }t j d|| t j d }||d d d f |d d d f ||f }t
|| d S )Nr r' )r r r r r zaabb->abr r )r+ randomnormalzerosr, r. r )r1 rw y1idxy2s r5 test_einsum_fixed_collapsingbugz*TestEinsum.test_einsum_fixed_collapsingbug s
I Q< 0 0
Xf
*aR((((ill
s111d7|SD\33
4R r> c t j d }t j d| }| dd }t || d S )N)r^ r^ r^ r^ zijij->r r )axis1axis2)r+ rN
random_sampler, r r
)r1 tensorrw r s r5 "test_einsum_failed_on_p9_and_s390xz-TestEinsum.test_einsum_failed_on_p9_and_s390x* sb (()9:: Ih ' ' LLqL * * 0 0 2 21 r> c t j d }t j d d d d }t j d }d|d d d< t j d|||| t |j | t j d d d d }t j d|||| t |j | t j d d }t j d }d|d < t j d
t j }t j d ||| t |j | d S ) Nr r^ r r z mi,mi,mi->mr z im,im,im->m)r r r ).r )r r ij,jk->ik)r+ r0 r, r r rG )r1 rw r correct_bases r5 (test_einsum_all_contig_non_contig_outputz3TestEinsum.test_einsum_all_contig_non_contig_output2 s+
GFOOgbkk##A# wr{{ SSqS
-Aqc222238\222gbkk##A#
-Aqc222238\222 gi (wy)) V GFBJ ' '
+q!----38\22222r> r AllFloat
AllIntegerc t j | }t j d dz | }t j d| }t j d| }|| k sJ t j d|| }t |||z t j d| d d d d | d d d d }t |||z t j d|| ||z k sJ t j d| }t j d||
d | }t |||
d z t j d
|| }t |||
d z t j d || }|t j d||z k sJ t j d || }|t j d||z k sJ t j g d
| }t j d||| }t |||z |z t j d|||| }t |||z |z |z d S )Nr ?r r` r zi,i->izi,i->r r r r r )rb rb g ?g @r zi,i,i->z i,i,i,i->) r+ r r. r rh r, r r repeatr0 type)r1 r arrscalarresr s r5 test_different_pathsz TestEinsum.test_different_pathsI s
y||c!))%00!5))) is##cggii i#s++3c ***i#**Q--!"4cjjmmCCaC6HII3c ***y#s++c /@/@@@@@gau%%%iejjmm===3ejjmm 3444i--3ejjmm 3444i,,biv|444444iV,,biv|444444h000>>>i 3S113sS 5 5 7 7888i S#sC883sS 3!6 ; ; = =>>>>>r> c @ t j dt j d d }d|d< t j dt j d d }t j dt j }t j d||| }t || d S ) N)r r' r' r` r T.)r r' r' r r )r+ rP bool_r0 r, r )r1 r3 r r rg s r5 test_small_boolean_arraysz$TestEinsum.test_small_boolean_arraysw s
HZrx 0 0 0! 4#hz2222A26girx000i,a<<<S# r> c t j d dd }t j d||| }||u sJ d S )Nro r
r r )r+ r. r/ r, )r1 r3 rg s r5 test_out_is_reszTestEinsum.test_out_is_res sI IaLL A & &i,a:::axxxxxxr> Nc |i|g}| d d d }|D ]4}d |D }| t j j | 5n|g|z }t j |ddi}t j |ddi}t
|| t j |ddi}t
|| d S ) N->r r" c ( g | ]}t | S r= )global_size_dictrz s r5 r] z/TestEinsum.optimize_compare.. s :::(+:::r> r Fr optimal)splitappendr+ rN randr, r ) r1
subscriptsoperandsargstermstermr@ nooptr? s r5 optimize_comparezTestEinsum.optimize_compare s c Z | d | d d S )Nz
a,ab,abc->abcz
a,b,ab->abr| r s r5 test_hadamard_like_productsz&TestEinsum.test_hadamard_like_products s0 o... l+++++r> c | d | d | d d S )Nzea,fb,gc,hd,abcd->efghzea,fb,abcd,gc,hd->efghzabcd,ea,fb,gc,hd->efghr~ r s r5 test_index_transformationsz%TestEinsum.test_index_transformations sG 6777 6777 677777r> c V | d | d | d | d | d | d | d | d d S )N$acdf,jbje,gihb,hfac,gfac,gifabc,hfacz cd,bdhe,aidb,hgca,gc,hgibcd,hgaczabhe,hidj,jgba,hiab,gabz bde,cdh,agdb,hica,ibd,hgicd,hiacz chd,bde,agbc,hiad,hgc,hgi,hiad chd,bde,agbc,hiad,bdi,cgh,agdbzbdhe,acad,hiab,agac,hibdr~ r s r5 test_complexzTestEinsum.test_complex s DEEE DEEE @AAA 7888 @AAA >??? >??? 899999r> c | d | d | d | d | d | d d S )Nz ab,ab,c->z
ab,ab,c->cz
ab,ab,cd,cd->zab,ab,cd,cd->aczab,ab,cd,cd->cdzab,ab,cd,cd,ef,ef->r~ r s r5
test_collapsezTestEinsum.test_collapse s k*** l+++ o... /000 /000 344444r> c | d | d | d | d | d | d d S )Nzab,cd,ef->abcdefzab,cd,ef->acdfzab,cd,de->abcdez ab,cd,de->bezab,bcd,cd->abcdzab,bcd,cd->abdr~ r s r5 test_expandzTestEinsum.test_expand s 0111 ./// /000 n--- /000 ./////r> c | d | d | d | d | d | d | d | d | d | d
| d | d d S )
N
eb,cb,fb->cefdd,fb,be,cdb->cefbca,cdb,dbf,afc->dcc,fce,ea,dbf->abzfdf,cdd,ccd,afe->aezabcd,adzed,fcd,ff,bcf->bezbaa,dcf,af,cde->bezbd,db,eac->acezfff,fae,bef,def->abdzefc,dbc,acf,fd->abez
ba,ac,da->bcdr~ r s r5 test_edge_caseszTestEinsum.test_edge_cases s o... 1222 1222 2333 3444 i((( 1222 2333 ./// 4555 3444 o.....r> c | d | d | d | d | d d S )Nzab,abzab,bazabc,abczabc,baczabc,cbar~ r s r5 test_inner_productz TestEinsum.test_inner_product sl g&&& g&&& i((( i((( i(((((r> c | d | d | d | d | d | d | d | d | d | d
| d d S ) Nzaab,fa,df,ecc->bdezecb,fef,bad,ed->aczbcf,bbb,fbf,fc->z bb,ff,be->ezbcb,bb,fc,fff->zfbb,dfd,fc,fc->zafd,ba,cc,dc->bfzadb,bc,fa,cfc->dzbbd,bda,fc,db->acfzdba,ead,cad->bcezaef,fbc,dca->bder~ r s r5 test_random_casesz TestEinsum.test_random_cases s 2333 2333 0111 m,,, /000 /000 0111 0111 2333 0111 011111r> c t j d ddddd }t j d| }t |dg d S )Nro r' r
zbbcdc->dr )r+ r. r/ r, r r5 s r5 test_combined_views_mappingz&TestEinsum.test_combined_views_mapping sN IaLL Aq!Q / / Ij! $ $Q r> c t j ddd }t j dd }t j dd }t j d }| d|||g | d||||g t j dddd }t j d d }| d
|||g | d ||||g t j d d
dd }| d||g d S )Nr' r r r r^ z ijk,kl,jl)rw zijk,kl,jl,i->ir z
abjk,kl,jlzabjk,kl,jl,ab->abr r r z obk,ijk->ioj)r+ rN ru r| r. r/ )r1 r3 r4 r r) efgs r5 test_broadcasting_dot_casesz&TestEinsum.test_broadcasting_dot_cases sL
INN1a # # INN1a INN1a INN2 kQ1I >>> .!Q1 FFF INN1aA & & INN1a laAY ??? 1Q1aL III
IbMM ! !!Q * * n1v >>>>>r> c t j dd }t j dd }dD ]6}t j d||d| }t |j j t j d||d | }t |j j t j d||d
| }t |j j t j d||d | }t |j j d u t |j j d u t j d|||
}t |j j d u t |j j d u 8t j dd }dD ]4}t j d||d| }t |j j 5t j dd }dD ]4}t j d||d| }t |j j 5d S )N)r r
r F)r r8 r z...ft,mf->...mtr3 r r r kFr C)r+ r0 r, r r f_contiguous c_contiguous)r1 r3 r4 r? rM r r) s r5 test_output_orderz TestEinsum.test_output_order s
GIS ) ) ) GF# & & & 5 5C) -q!3MMMC CI* + + +) -q!3MMMC CI* + + +) -q!3MMMC CI* + + +) -q!3MMMC CI*e3 4 4 4 CI*e3 4 4 4) -q!cBBBC CI*e3 4 4 4 CI*e3 4 4 4 4 GF# & & & , ,C) -q!3MMMC CI* + + + + GIS ) ) ) , ,C) -q!3MMMC CI* + + + + , ,r> )FrC )4r: r; r< r6 rj r r+ _no_nep50_warningr r r r r r r r r r r r r r r r r r r+ r6 rE rL rT rZ r^ r' mark parametrize typecodesrh rk rm r| r r r r r r r r r r r r= r> r5 r r s S6 S6 S6j== == ==~I. I. I.V Ro! o! o! o!b % % %% % %% % %% % %+ + ++ + +% % %% % %% % %% % %+ + +. . .+ + +& & &/ / // / /$B $B $BL
8
8
8 e e e(7 (7 (7T < 3 3 3. [ W
\*
% \(B
BD D*? *?D D*?X
( ( ( ($, , ,
8 8 8 : : :5 5 50 0 0
/
/
/ ) ) ) 2 2 2 ? ? ?(", ", ", ", ",r> r c H e Zd ZefdZd Zd Zd Zd Zd Z d Z
d Z d Z d
S ) TestEinsumPathc |g}| d d d }|D ]6}fd|D }| t j j | 7|S )Nro r r" c g | ]
}| S r= r= )r\ rw size_dicts r5 r] z1TestEinsumPath.build_operands..* s ///QIaL///r> )rs rt r+ rN ru )r1 stringr rw ry rz r@ s ` r5 build_operandsz TestEinsumPath.build_operands$ s{ 8 T""1%++C00 3 3D////$///D OOBIND 1 2 2 2 2r> c > t | t | k }t | t t | dz
D ]>}|t ||dz t z }|||dz ||dz k z }?t | d S )Nr' )lenr ri
isinstancetuple)r1 comp benchmarkretposs r5 assert_path_equalz TestEinsumPath.assert_path_equal/ s 4yyC NN* TQ '' 9 9C :d37mU33 3C DqMYsQw%77 8CC r> c | d }t j |ddi\ }}| |ddg t j |ddi\ }}| |ddg | d }t j |ddi\ }}| |ddg t j |ddi\ }}| |ddg d S ) Nz
a,b,c->abcr )r r einsum_pathr )rr r zacdf,jbje,gihb,hfacr r' r r
r r+ r r )r1
outer_testpathpath_str long_tests r5 test_memory_contraintsz%TestEinsumPath.test_memory_contraints8 s (( 66
LmLLh tmY%?@@@ MnMMh tmY%?@@@''(=>> K]KKh tm\%BCCC L^LLh tm\%BCCCCCr> c | d }t j |ddi\ }}| |g d t j |ddi\ }}| |g d | d }t j |ddi\ }}| |g d t j |ddi\ }}| |g d d S ) Nr r r )r )r
r r- )r r rl r r r r' rr r )r r- r r
r- r' r
r r )r )r r )r' r r- r r r r )r1
long_test1r r
long_test2s r5 test_long_pathsz TestEinsumPath.test_long_pathsJ sZ (()OPP
GhGGh t &V &V &V W W W HiHHh t &V &V &V W W W (()IJJ
GhGGh t &V &V &V W W W HiHHh t &V &V &V W W W W Wr> c V | d }t j |ddi\ }}| |g d t j |ddi\ }}| |g d | d }t j |ddi\ }}| |g d t j |ddi\ }}| |g d | d }t j |ddi\ }}| |g d t j |ddi\ }}| |g d | d
}t j |ddi\ }}| |g d t j |ddi\ }}| |g d | d d
d
d
d
d }t j |ddi\ }}| |g d t j |ddi\ }}| |g d d S )Nr r r )r r r rr r )r r r r r )r r r r r r r r r za,ac,ab,ad,cd,bd,bc->r )r3 r4 r r) )r )r r )r r' r r
r r r )r1
edge_test1r r
edge_test2
edge_test3
edge_test4s r5 test_edge_pathsz TestEinsumPath.test_edge_pathsa s (( 99
GhGGh t%D%D%DEEE HiHHh t%D%D%DEEE (()<==
GhGGh t%L%L%LMMM HiHHh t%L%L%LMMM (()<==
GhGGh t%L%L%LMMM HiHHh t%L%L%LMMM (()=>>
GhGGh t%L%L%LMMM HiHHh t%L%L%LMMM (()@9;"2TV3W3W ) Y Y
GhGGh t%P%P%PQQQ HiHHh t%P%P%PQQQQQr> c | d }t j |ddi\ }}| |ddg t j |ddi\ }}| |g d g d}t j |d|i\ }}| || t j |ddi}t j |d|i}t || d S ) Nr r Fr r Tr )r r r r r r+ r r r, r )r1 path_testr r exp_pathr{ r? s r5 test_path_type_inputz#TestEinsumPath.test_path_type_input s ''(<== CUCCh tm\%BCCC BTBBh t%L%L%LMMM::: FXFFh tX... 95u55i6X66E3'''''r> c | d }g d}t j |d|i\ }}| || t j |ddi}t j |d|i}t || d S )Nz cab,cdd->abr )r' r r Fr )r1 r r r r r{ r? s r5 #test_path_type_input_internal_tracez2TestEinsumPath.test_path_type_input_internal_trace s ''
66 000 FXFFh tX... 95u55i6X66E3'''''r> c j | d }g d}t t t j g|R d|i t t t j g|R d|i | d }g d}t t t j g|R d|i t t t j g|R d|i d S )Nzab,bc,cd,de->ae)r rl r r za,a,a->ar )r r RuntimeErrorr+ r, r )r1 r r s r5 test_path_type_input_invalidz+TestEinsumPath.test_path_type_input_invalid s ''(9:: 222lBIM MMMHMMM ". I+4 I I I?G I I I ''
33 000lBIM MMMHMMM ". I+4 I I I?G I I I I Ir> c t j dgg }t j ddgd D ] }t j dj | | d S )Nr' r r )rc z{}...a{}->{}...a{})r+ rh itertoolsproductr, format)r1 re sps r5 test_spaceszTestEinsumPath.test_spaces sa huoo#RIa888 = =B I1*126 < < < < = =r> N)
r: r; r< rq r r r r r r r r r r= r> r5 r r # s /? D D D$W W W.*R *R *RX( ( (& ( ( ( I I I= = = = =r> r c t j dt dd } t j dt dd }t j | | }t j d| | }t
|| t j d| || }t
|| d S )Nro r` r
r\ r )r+ r. intr/ r r, r )r3 r4 r) r s r5 test_overlapr s
!3 ''1--A
!3 ''1--A
q! A
+q!$$AA
+q!+++AAr> )r r' numpyr+
numpy.testingr r r r r r r r
charsrh sizesdictziprq r r r r= r> r5 r s@
222334E5))** P, P, P, P, P, P, P, P,d \= \= \= \= \= \= \= \=| r>