U
|
O ±dqÈ ã @ s d dl Z d dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
m Z m Z dZ e dddddddddddg ¡Zeee eZG d d
|
d
|
ZG d d d Zd d ZdS )é N)Úassert_Ú assert_equalÚassert_array_equalÚassert_almost_equalÚ assert_raisesÚsuppress_warningsÚassert_raises_regexÚassert_allcloseZ
|
abcdefghijé é é é é c @ sv e Zd Zdd Zdd Ze ¡ dVddZdd Zd
|
d Z d d Z
|
dd Z dd Z dd Z dd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Z d2d3 Z e j d4ej!d5 ej!d6 ¡d7d8 Z"d9d: Z#d;d< Z$dWd>d?Z%d@dA Z&dBdC Z'dDdE Z(dFdG Z)dHdI Z*dJdK Z+dLdM Z,dNdO Z-dPdQ Z.dRdS Z/dTdU Z0d=S )XÚ
|
TestEinsumc
|
C sj dD ]^}t ttj|d t ttjd|d t ttjdd|d t ttjddd|d t ttjddd|d t ttjddd |d
|
t ttjddd |d t ttjddd|d t ttjfdd|i t ttjddd|d t ttjdddgdg|d t ttjddg|d t ttjdd|d t ttjdddg|d t ttjdd|d t ttjdddg|d t ttjdd|d t ttjdddg|d t ttjdddg|d t ttjdddg|d t ttjdddg|d t ttjdddg|d t ttjdddg|d t ttjd ddg|d t ttjd ddg|d t ttjd ddg|d t ttjd ddgddgg|d t ttjd t d!¡ d"d#¡|d t ttjd$t d!¡ d"d#¡|d t ttjdt d!¡ d"d#¡|d t ttjd%dd&gdd&ggt d'¡ d"d"¡|d ttd(( t d)¡}t d*¡}t d+||¡ W 5 Q R X t ttjd%t d!¡ d,d&¡|d-d. qd S )/N©TF©ÚoptimizeÚ r Útest)Úoutr ÚW©Úorderr Zblah)Úcastingr Z bad_data_type)Údtyper )Zbad_argr )?NNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNr ú,ÚiÚijz...izi...jzi...zij...zi..z.i...zj->..jzj->.j...zi%...z...j$zi->&zi->ijzij->jijÚiir r
|
r úii->izi->ié r z'b')r r r r r )r r r z aabcb,abcéÿÿÿÿÚd)r r ) r Ú
|
ValueErrorÚnpÚeinsumÚ TypeErrorÚarangeÚreshaper Úones)ÚselfÚdo_optÚaÚb© r. úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_einsum.pyÚtest_einsum_errors s
|
ÿÿÿÿÿÿÿ
|
ÿ
|
ÿÿ ÿ
|
|
ÿz TestEinsum.test_einsum_errorsc s dD ]v}t d¡ d _t jd |d}t|j k t j tg|d}t|j k t jd |d}t|j k t| t j ddg|d}t|j k t| t jd |d}t|jd d
|
jd < t jd |d}t|jd t d¡ d _t jd |d}t|j k t| j t j ddg|d}t|j k t| j t d ¡ d _t jd |d}t|j k t| fddt
|
dD t j ddgdg|d}t|j k t| fddt
|
dD t d¡ d _t jd |d}t|j k t|dd D t j tddgtdg|d}t|j k t|dd D t jd |d}t|j k t|dd ddd¡D t j ddtgtdg|d}t|j k t|dd ddd¡D t jd |d}t|j k t| fd dt
|
dD t j tddgdtg|d}t|j k t| fd dt
|
dD t jd |d}t|j k t| fd dt
|
dD t j dddgddg|d}t|j k t| fd!dt
|
dD t jd" |d}t|j k t| fd#dt
|
dD t j ddtgdtg|d}t|j k t| fd$dt
|
dD t jd% |d}t|j k t| fd&dt
|
dD t j dtdgdtg|d}t|j k t| fd'dt
|
dD t jd( |d}t|j k t|d)d ddd¡D t j dtdgtdg|d}t|j k t|d*d ddd¡D t d¡ d _t jd+ |d}t|j k t| fd,dt
|
dD t j dddgdg|d}t|j k t| fd-dt
|
dD t d.¡ d/ _t jd0 |d}t|j k t| dd¡ t j dddgdddg|d}t|j k t| dd¡ qd S )1Nr r ©r
|
r z...r r r r Z WRITEABLEFZjié )r r r c s g | ]} ||f qS r. r. ©Ú.0r ©r, r. r/ Ú
|
<listcomp> s z0TestEinsum.test_einsum_views.<locals>.<listcomp>r c s g | ]} ||f qS r. r. r3 r5 r. r/ r6 s é )r r r z ...ii->...ic s" g | ] fd dt dD qS )c s g | ]} ||f qS r. r. r3 ©Úxr. r/ r6 s ú;TestEinsum.test_einsum_views.<locals>.<listcomp>.<listcomp>r ©Úrange©r4 r. r8 r/ r6 s c s" g | ] fd dt dD qS )c s g | ]} ||f qS r. r. r3 r8 r. r/ r6 ¤ s r: r r; r= r. r8 r/ r6 ¤ s z ii...->...ic s" g | ] fd dt dD qS )c s g | ]} ||f qS r. r. r3 r8 r. r/ r6 ¨ s r: r r; r= r. r8 r/ r6 ¨ s ÿr
|
c s" g | ] fd dt dD qS )c s g | ]} ||f qS r. r. r3 r8 r. r/ r6 s r: r r; r= r. r8 r/ r6 s ÿz ...ii->i...c s g | ]} d d
||f qS ©Nr. r3 r5 r. r/ r6 ² s c s g | ]} d d
||f qS r> r. r3 r5 r. r/ r6 ¶ s zjii->ijc s g | ]} d d
||f qS r> r. r3 r5 r. r/ r6 º s c s g | ]} d d
||f qS r> r. r3 r5 r. r/ r6 ¾ s z ii...->i...c s( g | ] } d dd¡dd
||f qS ©r
|
r r N©Ú transposer3 r5 r. r/ r6 Â s c s( g | ] } d dd¡dd
||f qS r? r@ r3 r5 r. r/ r6 Æ s z i...i->i...c s( g | ] } d dd¡dd
||f qS ©r r r
|
Nr@ r3 r5 r. r/ r6 Ê s c s( g | ] } d dd¡dd
||f qS rB r@ r3 r5 r. r/ r6 Î s z i...i->...ic s" g | ] fd dt dD qS )c s g | ]} ||f qS r. r. r3 r8 r. r/ r6 Ò s r: r r; r= r. r8 r/ r6 Ò s ÿc s" g | ] fd dt dD qS )c s g | ]} ||f qS r. r. r3 r8 r. r/ r6 × s r: r r; r= r. r8 r/ r6 × s ÿziii->ic s g | ]} |||f qS r. r. r3 r5 r. r/ r6 à s c s g | ]} |||f qS r. r. r3 r5 r. r/ r6 ä s é )r
|
r r zijk->jik) r$ r' Úshaper% r ÚbaseÚEllipsisr ÚflagsÚTr< rA Zswapaxes)r* r+ r- r. r5 r/ Útest_einsum_viewsg sÌ
|
|
|
|
|
|
|
|
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
|
z TestEinsum.test_einsum_viewsFc C sê t |¡}tddD ]b}t j||d}tt jd||dt j|dd |¡ tt j|dgg |dt j|dd |¡ qtddD ]t}t jd | |d d
|
d |¡}tt jd ||dt j|dd |¡ tt j|t dgt g|dt j|dd |¡ qtddD ]t}t jd
|
| |d d
|
|¡}tt jd ||dt j|dd |¡ tt j|dt gt g|dt j|dd |¡ qtddD ]v}t jd | |d d
|
d |¡}tt jd ||dt j|dd |¡ tt j|dt gt g|dt j|dd |¡ qtddD ]¾}t j|| |d ||¡}tt jd||dt
|
|¡ |¡ tt j|ddg|dt
|
|¡ |¡ t ddg¡}tt j|||dt
|
|¡ |¡ tt j|t ||dt
|
|¡ |¡ qtt dd d¡d tddD ]}t jd | |d d |¡}t jd | |d d
|
d |¡}tt jd|||dt ||¡ tt j|t g|t g|dt ||¡ qâtddD ]z}t jd | |d d
|
d |¡}t j||d}tt jd|||dt ||¡ tt j|t dg|t dg|dt ||¡ qptddD ]}t j|d d
|
|d |d d
|
¡}t j||d}tt jd|||dt |j|j¡j tt j|dt g|dt g|dt |j|j¡j qötddD ]p}t jd |dd }t j||dd }tt jd|||dt ||¡ tt j|dg|dg|dt ||¡ qt â}| t j¡ tddD ] }t jd| |d d|¡}t j||d}tt jd|||dt ||¡ tt j|ddg|dg|dt ||¡ t jd|d}t jd|||dd|d t|t | d¡| d¡¡ |¡ d|d< t j|ddg|dg|dd|d t|t | d¡| d¡¡ |¡ q tddD ],}t jd| |d d|¡}t j||d}tt jd |j|j|dt |j|j¡ tt j|jddg|jdg|dt |j|j¡ t jd|d}t jd |j|j|dd|d t|t |j d¡|j d¡¡ |¡ d|d< t j|jddg|jdg|dd|d t|t |j d¡|j d¡¡ |¡ q8tddD ]}|d k s|d krrt jd| |d d|¡}t j|d |d |d ¡}tt jd |||dt ||¡ tt j|ddg|dd
|
g|dt ||¡ qrtddD ]Ú}t jd| |d d|¡}t j|d |d |d ¡}t jd |d dd ¡}t jd |||dd|d t|t | d¡| d¡¡ |¡ d|d< t j|ddg|dd
|
g|dd|d t|t | d¡| d¡¡ |¡ qt jd|d d d¡}t jd!|d dd"¡}t jd#|d d"d ¡}|d k rtt jd$||||d| |¡ |¡ tt j|ddg|dd
|
g|d
|
d g|d| |¡ |¡ t jd%|d d d ¡} t jd$|||| dd|d | d¡ | d¡¡}
|
|
|
| d¡¡ |¡}
|
t| |
|
d| d< t j|ddg|dd
|
g|d
|
d g| dd|d
|
| d¡ | d¡¡}
|
|
|
| d¡¡ |¡}
|
t| |
|
t |¡t d ¡k rät jd&|d d dd"¡}t jd |d dd d
|
¡}tt d'||¡t j||ddgddgfd( tt |ddd
|
g|ddd gd
|
d g¡t j||ddgddgfd( t jd)|d d"d
|
¡}t jd*|||dd|d t|t j| d¡| d¡ddgddgfd( |¡ d|d< t j|ddd
|
g|ddd gd
|
d g|dd|d t|t j| d¡| d¡ddgddgfd( |¡ W 5 Q R X |jd+k rþd,nt |¡jd } t jdd | ddd-ddg|d}t jdd.d/| ddd dg|d}t d0d0d1d0d0d1d0d0g¡}tt jd2|||d3d|d4t t |dk|dk¡|dk¡ tt j|dg|dg|dgdgd3dd5 t t |dk|dk¡|dk¡ t jd6|d}tt d7d |¡d t |¡ tt d g |dgg ¡d t |¡ tt d8|d ¡d t |¡ tt |dgd g g ¡d t |¡ tdd9D ] }t j||d}t |¡jdk rtt jd:|||dt ||¡ tt jd;|||dt ||¡ tt jd<|d
|
|dd
|
| tt jd=d
|
||dd
|
| tt jd8|d
|
|dd
|
t |¡ tt jd7d
|
||dd
|
t |¡ tt jd:|dd
|d d
|dt |dd
|d d
¡ tt jd;|dd
|d d
|dt |dd
|d d
¡ tt jd<|dd
d
|
|dd
|
|dd
tt jd=d
|
|dd
|dd
|
|dd
tt jd8|dd
d
|
|dd
|
t |dd
¡ tt jd7d
|
|dd
|dd
|
t |dd
¡ qt jd6t d}t jd||dd5}t|t |¡ t|jt |¡ t j|dgg |dd5}t|t |¡ t|jt |¡ t d
|
¡d } t d¡ d
|
d
|
¡d } t d¡ d
|
d
|
¡d> }tt d?| | |¡d@ t dA¡} t dB¡} t t jdC| | d0dt jdC| | d1d t t jdC| | d0ddDgd
|
t dEdFg¡}t dGg¡}t t jdH||d1ddI t t jdH||d0ddI t dJ¡d
|
} t dK¡d
|
} dLD ]L}t t jdM| | |dt jdM| | |d t t jdM| | |dt dJdN¡ q.t j d
|
|d}t j d
|
|d}t t jdO|||ddEg t t jdP|||ddEg t t jdQ|||ddEg d S )RNr é ©r úi->r r! )Zaxisr r r
|
r z ...i->...z i...->...r z..., ...r é z
|
...i, ...ié z
|
i..., i...úi,jzij, jzij,jÚf8Úunsafe)r r r r .zji,jé Úf2zij,jkrC é r é zij,jk,klé é< zijk, jil -> kl©Zaxesé
|
z ijk,jil->klÚuéþÿÿÿé g @g TFzi,i,i->iú?)r r r )r r r2 ú,i->úi,->é z...,...zi,iúi,->iú,i->ié z z,mz,zm->éý )rY r
|
©r r
|
zij,ij->jg $@g @ç @g @zi, ig 4@)r r ©r r r ú...ij,...jk->...ikg ô?zji,i->zi,ij->zij,i->)!r$ r r< r' r r% ÚsumÚastyper( rF ÚtraceZasarrayÚlistÚmultiplyÚinnerrH Úouterr ÚfilterZComplexWarningÚdotÚ tensordotÚkindZiinfoÚmaxÚarrayÚ logical_andÚitemsizeÚobjectr) r ÚfullZeye)r* r r+ Únr, Znp_arrayr- ÚsupÚcr" ÚtgtZneg_valÚpÚqÚrr9 Úyr r. r. r/ Úcheck_einsum_sumsò sº
|
ÿÿ ÿÿÿÿ ÿÿÿÿÿÿ
|
ÿ
|
ÿ
|
ÿ ÿ ÿ
|
ÿ
|
ÿ
|
|
ÿ
|
ÿ ÿ ÿÿÿ ÿ ÿÿÿÿ ÿ ÿ
|
ÿÿÿ ÿ
|
ÿÿÿ
|
ÿ
|
ÿÿ ÿÿÿ ÿ ÿÿÿ
|
ÿ ÿÿ ÿ
|
ÿ
|
ÿ"ÿ ÿÿÿ ÿÿÿ ÿ þ ÿ þ $ $
|
ÿ """ÿ"ÿ,,ÿÿ
|
|
ÿÿ ÿ
|
ÿþ ÿ
|
þÿÿÿz TestEinsum.check_einsum_sumsc C s | d¡ d S )NÚi1©r ©r* r. r. r/ Útest_einsum_sums_int8 s z TestEinsum.test_einsum_sums_int8c C s | d¡ d S )NÚu1r r
r. r. r/ Útest_einsum_sums_uint8 s z!TestEinsum.test_einsum_sums_uint8c C s | d¡ d S )NÚi2r r
r. r. r/ Útest_einsum_sums_int16# s z!TestEinsum.test_einsum_sums_int16c C s | d¡ d S )NÚu2r r
r. r. r/ Útest_einsum_sums_uint16& s z"TestEinsum.test_einsum_sums_uint16c C s | d¡ | dd¡ d S )NÚi4Tr r
r. r. r/ Útest_einsum_sums_int32) s
|
z!TestEinsum.test_einsum_sums_int32c C s | d¡ | dd¡ d S )NZu4Tr r
r. r. r/ Útest_einsum_sums_uint32- s
|
z"TestEinsum.test_einsum_sums_uint32c C s | d¡ d S )NÚi8r r
r. r. r/ Útest_einsum_sums_int641 s z!TestEinsum.test_einsum_sums_int64c C s | d¡ d S )NÚu8r r
r. r. r/ Útest_einsum_sums_uint644 s z"TestEinsum.test_einsum_sums_uint64c C s | d¡ d S )NrS r r
r. r. r/ Útest_einsum_sums_float167 s z#TestEinsum.test_einsum_sums_float16c C s | d¡ d S )NZf4r r
r. r. r/ Útest_einsum_sums_float32: s z#TestEinsum.test_einsum_sums_float32c C s | d¡ | dd¡ d S )NrP Tr r
r. r. r/ Útest_einsum_sums_float64= s
|
z#TestEinsum.test_einsum_sums_float64c C s | tj¡ d S r> )r r$ Z
|
longdoubler
r. r. r/ Útest_einsum_sums_longdoubleA s z&TestEinsum.test_einsum_sums_longdoublec C s | d¡ | dd¡ d S )NZc8Tr r
r. r. r/ Útest_einsum_sums_cfloat64D s
|
z$TestEinsum.test_einsum_sums_cfloat64c C s | d¡ d S )NZc16r r
r. r. r/ Útest_einsum_sums_cfloat128H s z%TestEinsum.test_einsum_sums_cfloat128c C s | tj¡ d S r> )r r$ Z clongdoubler
r. r. r/ Ú test_einsum_sums_clongdoubleK s z'TestEinsum.test_einsum_sums_clongdoublec C s
|
t d¡}t d¡}tt d||¡dgdggg tt jd||dddgdggg tt d||¡dgdggg tt dddd gdd d
|
g¡d tt jdddd gdd d
|
gd dd t d t j¡}t dt j¡}tt d|||¡t d||¡ tt jd|||ddt jd||dd t dd ¡}t dd¡ dd¡}t dd¡ d
|
d¡}tt d|||¡dd gd dgddgddggddgddgddgdd ggg tt jd|||dddd gd dgddgddggddgddgddgdd ggg tt jd dgdgd d dgg d S ) Nre )r
|
r
|
r zij...,j...->i...r
|
Tr z ...i,...ir r r rT Úgreedy)r rM r r
|
r )r rM rN zijklm,ijn,ijn->z ijklm,ijn->r r2 z x,yx,zx->xzyé rc é rR é é rC é0 é@ rO ©r )r$ r) r r% Úint64r' r( )r* r, r- r| r. r. r/ Útest_einsum_miscN sB
|
|
" ÿÿ ÿÿ ÿÿÿÿzTestEinsum.test_einsum_miscc s¦ t d¡ t d¡t j ddgddgddgdd t j ddgddgddgdd t j dd gd dgddgdd tt fd
|
d tt fd d d S ) Nr1 ©r r r rT r
|
Fr r7 é3 c s" t j ddgddgddgddS )Nr é4 r
|
Fr ©r$ r% r. ©r, r- r. r/ Ú<lambda>| ó z1TestEinsum.test_subscript_range.<locals>.<lambda>c s" t j ddgddgddgddS )Nr! r r
|
Fr r¨ r. r© r. r/ rª } r« )r$ r) r% r r# r
r. r© r/ Útest_subscript_ranget s
|
|
"""z TestEinsum.test_subscript_rangec C s< t d¡ ddd¡}t d¡}t jd||dd}dD ]L}tt jd |||d| tt jd
|
|||d| tt jd |||d| q4t d ¡ d ¡}t d¡ d¡}t jd||dd}dD ]d}tt jd|||d| tt jd|||d| tt jd|||d| tt jd|||d| q¸ddddg}t t |¡¡ |¡}t |d ¡}t jd||dd}dD ]N}tt jd|||d| tt jd|||d| tt jd|||d| qdd\}} }
|
t || |
|
¡ ddd|| |
|
¡}t || |
|
d ¡ || |
|
d¡}t jd ||dd}dD ] }tt jd |||d| qd S ) NrC r
|
r r z
|
ijk,j->ijkFr r zij...,j...->ij...zij...,...j->ij...zij...,j->ij...rM ©r r r )r r
|
z ik,kj->ijzik...,k...->i...zik...,...kj->i...jz...k,kjz ik,k...->i...r z ijkl,k->ijlzijkl,kz...kl,kz
|
...kl,k...)é r® éx r z...lmn,...lmno->...oz...lmn,lmno->...o)r$ r' r( r% r Úprod) r* ÚAÚBÚrefÚoptÚdimsr, ÚvÚJÚKÚMr. r. r/ Útest_einsum_broadcast sB
|
|
"" ÿÿz TestEinsum.test_einsum_broadcastc C s t d¡ dd¡ t j¡}t d¡ ddd¡ t j¡}t d||¡}t j||dd}t|| t d ¡ dd¡ t j ¡}t d
|
¡ ddd d ¡ t j¡}t d ||¡}t j||dd}t|| d S ) Nr r
|
r i@ i«
|
z cl, cpx->lpx)r r rX r2 i r¡ zcl, cpxy->lpxy)
|
r$ r' r( rj Úfloat32Úint16r% rr r Úfloat64)r* r± r² ÚesÚtpr. r. r/ Útest_einsum_fixedstridebug© s
|
z%TestEinsum.test_einsum_fixedstridebugc C sj t j ddd¡}t d¡}t jd||d t d¡}||d d
d f |d d
d f ||f }t|| d S )Nr r )r r r r rg zaabb->abr¢ r )r$ ÚrandomÚnormalÚzerosr% r' r )r* r9 Úy1ÚidxÚy2r. r. r/ Ú test_einsum_fixed_collapsingbugÇ s
|
|
(z*TestEinsum.test_einsum_fixed_collapsingbugc C s8 t j d¡}t d|¡}|jddd ¡ }t|| d S )N)rY rY rY rY zijij->r r
|
)Zaxis1Zaxis2)r$ rÁ Z random_sampler% rk r )r* Ztensorr9 r r. r. r/ Ú"test_einsum_failed_on_p9_and_s390xÑ s z-TestEinsum.test_einsum_failed_on_p9_and_s390xc C sÚ t d¡}t d¡d d d
}t d¡}d|d d d
< t jd||||d t|j| t d¡d d d
}t jd||||d t|j| t d¡d }t d¡}d|d < t d
|
t j¡}t jd |||d t|j| d S ) Nrg rY r
|
r z mi,mi,mi->mr¢ z im,im,im->m)r
|
r
|
r
|
).r )r
|
r
|
ú ij,jk->ik)r$ r) r% r rE r» )r* r9 r Z correct_baser. r. r/ Ú(test_einsum_all_contig_non_contig_outputÙ s
|
|
|
z3TestEinsum.test_einsum_all_contig_non_contig_outputr ZAllFloatZ
|
AllIntegerc C sØ t |¡}t d¡d |¡}t jd|d}t d|¡}|| ¡ ksHtt d||¡}t||| t d| d¡d d d
| d¡d d d
¡}t||| t d||¡|| ¡ ks¾tt j
|
d|d}t jd|| d¡|d }t||| d¡ t d
|
||¡}t||| d¡ t d ||¡}|t d|| ¡ksBtt d ||¡}|t d|| ¡ksjtt jddd ddg|d}t d|||¡}t||| | ¡ t d||||¡}t||| | | ¡ d S )Nrc g à?r
|
rK rL zi,i->izi,i->ra r¢ rb r^ r_ g Ð?g @rf zi,i,i->z i,i,i,i->) r$ r r' rj ru r% ri ÚAssertionErrorr Úrepeatr) Útype)r* r ÚarrZscalarÚresr r. r. r/ Útest_different_pathsð s0
|
. z TestEinsum.test_different_pathsc C sh t jdt jdd d
}d|d< t jdt jdd d
}t jdt jd}t jd|||d}t|| d S ) N)r r r rK r
|
T.)r
|
r r rh r¢ )r$ rà Zbool_r) r% r )r* r, r r} rÏ r. r. r/ Útest_small_boolean_arrays s z$TestEinsum.test_small_boolean_arraysc C s4 t d¡ dd¡}t jd|||d}||ks0td S )Nr2 r rh r¢ )r$ r' r( r% rË )r* r, rÏ r. r. r/ Útest_out_is_res( s zTestEinsum.test_out_is_resNc C s |d krN|g}| d¡d d¡}|D ]$}dd |D }| tjj| ¡ q&n
|
|g| }tj|ddi}tj|ddi}t|| tj|dd i}t|| d S )
|
Nú->r r c S s g | ] }t | qS r. )Úglobal_size_dict©r4 r9 r. r. r/ r6 4 s z/TestEinsum.optimize_compare.<locals>.<listcomp>r Fr Úoptimal)ÚsplitÚappendr$ rÁ Úrandr% r ) r* Z
|
subscriptsÚoperandsÚargsÚtermsÚtermrµ Únooptr´ r. r. r/ Úoptimize_compare- s
|
|
zTestEinsum.optimize_comparec C s | d¡ | d¡ d S )Nz a,ab,abc->abcz
|
a,b,ab->ab©rß r
r. r. r/ Útest_hadamard_like_products? s
|
z&TestEinsum.test_hadamard_like_productsc C s" | d¡ | d¡ | d¡ d S )Nzea,fb,gc,hd,abcd->efghzea,fb,abcd,gc,hd->efghzabcd,ea,fb,gc,hd->efghrà r
r. r. r/ Útest_index_transformationsD s
|
|
z%TestEinsum.test_index_transformationsc C sT | 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
r. r. r/ Ú test_complexJ s
|
|
|
|
|
|
|
zTestEinsum.test_complexc C s@ | 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
r. r. r/ Ú test_collapseU s
|
|
|
|
|
zTestEinsum.test_collapsec C s@ | 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
r. r. r/ Ú test_expand^ s
|
|
|
|
|
zTestEinsum.test_expandc C s| | d¡ | d¡ | d¡ | d¡ | d¡ | d¡ | d¡ | d¡ | d ¡ | d
|
¡ | d ¡ | d ¡ d S ) Nú eb,cb,fb->cefúdd,fb,be,cdb->cefúbca,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
r. r. r/ Útest_edge_casesg s
|
|
|
|
|
|
|
|
|
|
|
zTestEinsum.test_edge_casesc C s6 | d¡ | d¡ | d¡ | d¡ | d¡ d S )Nzab,abzab,bazabc,abczabc,baczabc,cbarà r
r. r. r/ Útest_inner_productv s
|
|
|
|
|
z TestEinsum.test_inner_productc C sr | 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
r. r. r/ Útest_random_cases~ s
|
|
|
|
|
|
|
|
|
|
z TestEinsum.test_random_casesc C s4 t d¡ ddddd¡}t d|¡}t|dg d S )Nr2 r r zbbcdc->drM )r$ r' r( r% r )r* r, r- r. r. r/ Útest_combined_views_mapping s z&TestEinsum.test_combined_views_mappingc C sÖ t j ddd¡}t j dd¡}t j dd¡}t j d¡}| jd|||gd | jd||||gd t j dddd¡}t j d d ¡}| jd
|
|||gd | jd ||||gd t d ¡ d dd¡}| jd||gd d S )Nr r r r rY z ijk,kl,jl)rÚ zijk,kl,jl,i->irc z
|
abjk,kl,jlzabjk,kl,jl,ab->abr¡ r
|
rR z obk,ijk->ioj)r$ rÁ rÙ rß r' r( )r* r, r- r| r" ÚeÚfÚgr. r. r/ Útest_broadcasting_dot_cases s z&TestEinsum.test_broadcasting_dot_casesc C sb t jddd}t jddd}dD ]Ê}t jd||d|d}t|jj t jd||d |d}t|jj t jd||d
|
|d}t|jj t jd||d |d}t|jjd k t|jjd k t jd|||d }t|jjd k t|jjd k q t jddd}dD ]$}t jd||d|d}t|jj qþt jddd}dD ]&}t jd||d|d}t|jj q6d S )N)r
|
r r ÚF)r r r z...ft,mf->...mtr, r rñ r| ÚkFr ÚC)r$ r) r% r rG Ú f_contiguousÚ c_contiguous)r* r, r- r´ Útmpr| r" r. r. r/ Útest_output_order¦ s. z TestEinsum.test_output_order)F)N)1Ú__name__Ú
|
__module__Ú __qualname__r0 rI r$ Z_no_nep50_warningr r r r r r r r r r r r r r r r r¤ r¬ rº rÀ rÇ rÈ rÊ ÚpytestÚmarkZ parametrizeZ typecodesrÐ rÑ rÒ rß rá râ rå ræ rç rì rí rî rï ró rú r. r. r. r/ r s^ U ,& *
|
ÿ
|
,
|
|
r c @ sX e Zd ZefddZdd Zdd Zdd Zd d
|
Zd d Z d d Z
|
dd Z dd Z dS )ÚTestEinsumPathc sL |g}| d¡d d¡}|D ](} fdd|D }| tjj| ¡ q |S )NrÓ r r c s g | ] } | qS r. r. rÕ ©Ú size_dictr. r/ r6 Ñ s z1TestEinsumPath.build_operands.<locals>.<listcomp>)r× rØ r$ rÁ rÙ )r* Ústringr rÚ rÜ rÝ rµ r. r r/ Úbuild_operandsË s z TestEinsumPath.build_operandsc C sl t |t |k}t| tt |d D ]6}|t||d tM }|||d ||d kM }q(t| d S )Nr )Úlenr r< Ú
|
isinstanceÚtuple)r* ÚcompZ benchmarkÚretÚposr. r. r/ Úassert_path_equalÖ s z TestEinsumPath.assert_path_equalc C s¨ | d¡}tj|ddi\}}| |ddg¡ tj|ddi\}}| |ddg¡ | d¡}tj|ddi\}}| |ddg¡ tj|ddi\}}| |ddg¡ d S ) Nz
|
a,b,c->abcr )r r Ú einsum_path)r r r
|
)rÖ r zacdf,jbje,gihb,hfac©r r r
|
r ©r r$ r r )r* Z
|
outer_testÚpathÚpath_strZ long_testr. r. r/ Útest_memory_contraintsß s
|
|
z%TestEinsumPath.test_memory_contraintsc
|
C sÐ | d¡}tj|ddi\}}| |dddddd d
|
g¡ tj|dd i\}}| |dddddd d
|
g¡ | d ¡}tj|ddi\}}| |ddd dddd
|
g¡ tj|dd i\}}| |ddddddd
|
g¡ d S )Nrã r r r )r r r¥ )r
|
r r1 ©r r
|
©r r rÖ rä ©r r )r r re )r r )r r r )r* Z
|
long_test1r r Z
|
long_test2r. r. r/ Útest_long_pathsñ sL
|
ÿ ÿ
|
ÿ ÿz TestEinsumPath.test_long_pathsc C sÌ | d¡}tj|ddi\}}| |dddg¡ tj|ddi\}}| |dddg¡ | d¡}tj|ddi\}}| |dd ddg¡ tj|ddi\}}| |dd ddg¡ | d
|
¡}tj|ddi\}}| |dd ddg¡ tj|ddi\}}| |dd ddg¡ | d ¡}tj|ddi\}}| |dd ddg¡ tj|ddi\}}| |dd ddg¡ | j d dddddd}tj|ddi\}}| |dddg¡ tj|ddi\}}| |dddg¡ d S )Nrè r r r r r rÖ ré r rê re rë za,ac,ab,ad,cd,bd,bc->rT )r, r- r| r" r )r r r
|
r r r r )r* Z
|
edge_test1r r Z
|
edge_test2Z
|
edge_test3Z
|
edge_test4r. r. r/ Útest_edge_paths s6
|
|
|
|
ÿz TestEinsumPath.test_edge_pathsc C s° | d¡}tj|ddi\}}| |ddg¡ tj|ddi\}}| |ddddg¡ dd d dg}tj|d|i\}}| ||¡ tj|ddi}tj|d|i}t|| d S )
|
Nrë r Fr r Tre r r ©r r$ r r r% r )r* Ú path_testr r Úexp_pathrÞ r´ r. r. r/ Útest_path_type_input4 s
|
z#TestEinsumPath.test_path_type_inputc C sb | d¡}dddg}tj|d|i\}}| ||¡ tj|ddi}tj|d|i}t|| d S )Nz cab,cdd->abr ©r r r Fr )r* r r r r rÞ r´ r. r. r/ Ú#test_path_type_input_internal_traceG s
|
|
z2TestEinsumPath.test_path_type_input_internal_tracec C s | d¡}dddg}tttjf|d|i tttjf|d|i | d¡}dddg}tttjf|d|i tttjf|d|i d S )Nzab,bc,cd,de->aer r1 r r za,a,a->ar )r r Ú RuntimeErrorr$ r% r )r* r r r. r. r/ Ú test_path_type_input_invalidT s,
|
|
ÿÿÿ
|
|
ÿÿÿz+TestEinsumPath.test_path_type_input_invalidc C s< t dgg¡}tjddgddD ]}t dj| |¡ q d S )Nr r ú r )rÌ z{}...a{}->{}...a{})r$ ru Ú itertoolsÚproductr% Úformat)r* rÎ Úspr. r. r/ Ú test_spacesa s zTestEinsumPath.test_spacesN) rû rü rý rÔ r r r r r r r r r$ r. r. r. r/ r Ê s , r c C sp t jdtd dd¡} t jdtd dd¡}t | |¡}t d| |¡}t|| t jd| ||d}t|| d S )Nr2 rK r rÉ r¢ )r$ r' Úintr( rq r% r )r, r- r" r| r. r. r/ Ú test_overlaph s
|
r& )r rþ Únumpyr$ Z numpy.testingr r r r r r r r Úcharsru ÚsizesÚdictÚziprÔ r r r& r. r. r. r/ Ú<module> s" ( @
|
