
    /jЊ                    `    d dl mZ d dlZd dlZd dlmZ d dlmZ 	 d	dZ	 G d de          Z
dS )
    )annotationsN)Base)expectmeanc                   | j         }t          |          dk    rt          d          |j         }|d         }|d         }	t          j        | dd          }
t          j        | |
z
            }|t          j        |dd          z  }t          j        |          }d }|du rt          j        |          }d }|pt          j	        |t          j
        |t          j                  d          }|8t          j        ||k    d|                              t          j                  }n:|8t          j        ||k    dd                              t          j                  }t          |          d	k    r/|                    ||	d
f          }|                    |d
f          }|j         d         }t          j        ||ft          j                  }t#          |          D ]P}t#          |          D ]>}||         |         |k    r*||         ||         |                  |          ||         |<   ?Q|}t          |          d	k    r|                    |          }|>||z  }|dk    r3|                                |                                z  }|du r||fS |S |dk    rt          j        |          }n|dk    rt          j        |          }|r||fS |S )N   zUnsupported shaper   T)axiskeepdimsdtypeclip)mode      r   sum)shapelenRuntimeErrornpmaxexpr   logcopytakearrayint32whereastypefloat32reshapezerosranger   )xtargetweight	reductionignore_indexget_log_probinput_shapetarget_shapeNCmax_xexp_xpinplog_probgather_weightDneg_gather_element_inputidlosss                        t/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/softmaxcrossentropy.pysoftmaxcrossentropyr:      s     'K
;1.///<LAAAA F11t,,,EF1u9Eu1t4444A
&))CHt73<< M
 rx(H(H(HvVVV #HV|%;QNNUUj V  M 
	!<!7A>>EEBJEWW ;1kk1a*%%B(( 		!A!xAbjAAA1XX J Jq 	J 	JAay||++25a&12Fq2I1I(+A.	J $D ;1||L))  t#88:: 1 1 3 33Dt##X~%KFwt}}	e		vd|| X~K    c                  >   e Zd Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Z	ed&d	            Z
ed&d
            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Zed&d            Z ed&d             Z!ed&d!            Z"ed&d"            Z#ed&d#            Z$ed&d$            Z%d%S )'SoftmaxCrossEntropyLossreturnNonec                    d} t           j                            dddgdg|           }t          j                            d           t          j                            dd	                              t          j                  }t          j        	                    dd	d
                              t          j
                  }t          ||d          }t          |||g|gd           d S )Nnoner=   r$   yzinputsoutputsr'   r   r      r   highsizer'   test_sce_nonerE   rF   nameonnxhelper	make_noder   randomseedrandr   r    randintint64r:   r   r'   noder$   labelssces        r9   export_softmaxcrossentropy_nonez7SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_none]   s     	 {$$%:E	 % 
 
 		qINN1a  ''
33""114"88??II "!Vv>>> 	tQK#_MMMMMMr;   c                    d} t           j                            dddgddg|           }t          j                            d           t          j                            d	d
                              t          j                  }t          j        	                    dd
d                              t          j
                  }t          ||dd          \  }}t          |||g||gd           d S )NrA   r=   r$   rB   rC   r2   rD   r   r   rG   rH   rI   Tr'   r)   test_sce_none_log_probrN   rP   r'   rZ   r$   r[   r8   r2   s         r9   (export_softmaxcrossentropy_none_log_probz@SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_none_log_probu   s     	 {$$%:*%	 % 
 
 		qINN1a  ''
33""114"88??II -vd
 
 
h
 	v;8$)		
 	
 	
 	
 	
 	
r;   c                 $   d} t           j                            dg ddg|           }t          j                            d           t          j                            dd                              t          j                  }t          j        	                    ddd	
                              t          j
                  }t          j        g dt          j                  }t          |||d          }t          ||||g|gd           d S )NrA   r=   r$   rB   wrC   rD   r   r   rG   rH   rI   ?gffffff?g?rg   rg   r   r&   r'   test_sce_none_weightsrN   rQ   rR   rS   r   rT   rU   rV   r   r    rW   rX   r   r:   r   r'   rZ   r$   r[   weightsr\   s         r9   'export_softmaxcrossentropy_none_weightsz?SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_none_weights   s    	 {$$%"??E	 % 
 
 		qINN1a  ''
33""114"88??II(444BJGGG "!VGvNNN 	vw'E(		
 	
 	
 	
 	
 	
r;   c                 0   d} t           j                            dg dddg|           }t          j                            d           t          j                            dd	                              t          j                  }t          j        	                    dd	d
                              t          j
                  }t          j        g dt          j                  }t          |||dd          \  }}t          ||||g||gd           d S )NrA   r=   rd   rC   r2   rD   r   r   rG   rH   rI   rf   r   Tr&   r'   r)   test_sce_none_weights_log_probrN   rj   r'   rZ   r$   r[   rl   r8   r2   s          r9   0export_softmaxcrossentropy_none_weights_log_probzHSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_none_weights_log_prob   s     	 {$$%"??*%	 % 
 
 		qINN1a  ''
33""114"88??II(444BJGGG -vgd
 
 
h
 	vw'8$1		
 	
 	
 	
 	
 	
r;   c                    d} t           j                            dddgdg|           }t          j                            d           t          j                            dd	                              t          j                  }t          j        	                    dd	d
                              t          j
                  }t          ||d          }t          |||g|gd           d S )Nr   r=   r$   rB   rC   rD   r   r   rG   rH   rI   rL   test_sce_sumrN   rP   rY   s        r9   export_softmaxcrossentropy_sumz6SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_sum   s     	 {$$%:E	 % 
 
 		qINN1a  ''
33""114"88??II "!Vu=== 	tQK#^LLLLLLr;   c                    d} t           j                            dddgddg|           }t          j                            d           t          j                            d	d
                              t          j                  }t          j        	                    dd
d                              t          j
                  }t          ||dd          \  }}t          |||g||gd           d S )Nr   r=   r$   rB   rC   r2   rD   r   r   rG   rH   rI   Tr_   test_sce_sum_log_probrN   rP   ra   s         r9   'export_softmaxcrossentropy_sum_log_probz?SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_sum_log_prob   s     	 {$$%:*%	 % 
 
 		qINN1a  ''
33""114"88??II -vT
 
 
h
 	v;8$(		
 	
 	
 	
 	
 	
r;   c                    d} t           j                            dddgdg|           }t          j                            d           t          j                            dd	                              t          j                  }t          j        	                    dd	d
                              t          j
                  }t          ||          }t          |||g|gd           d S )Nr   r=   r$   rB   rC   rD   r   r   rG   rH   rI   test_sce_meanrN   rP   rY   s        r9   export_softmaxcrossentropy_meanz7SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean	  s     	 {$$%:E	 % 
 
 		qINN1a  ''
33""114"88??II "!V,, 	tQK#_MMMMMMr;   c                    d} t           j                            dddgddg|           }t          j                            d           t          j                            d	d
                              t          j                  }t          j        	                    dd
d                              t          j
                  }t          ||d          \  }}t          |||g||gd           d S )Nr   r=   r$   rB   rC   r2   rD   r   r   rG   rH   rI   Tr)   test_sce_mean_log_probrN   rP   ra   s         r9   (export_softmaxcrossentropy_mean_log_probz@SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_log_prob!  s     	 {$$%:*%	 % 
 
 		qINN1a  ''
33""114"88??II -QTJJJh 	v;8$)		
 	
 	
 	
 	
 	
r;   c                    d} t           j                            dddgdg|           }t          j                            d           t          j                            dd	d
                              t          j                  }t          j        	                    dd	d                              t          j
                  }t          ||          }t          |||g|gd           d S )Nr   r=   r$   rB   rC   rD   r   r   rG   r   r   r   rI   test_sce_mean_3drN   rP   )r'   rZ   r$   rB   r\   s        r9   "export_softmaxcrossentropy_mean_3dz:SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_3d>  s     	 {$$%:E	 % 
 
 		qINN1a##**2:66Iaaf55<<RXFF "!Q'' 	tQFSE8JKKKKKKr;   c                    d} t           j                            dddgddg|           }t          j                            d           t          j                            d	d
d                              t          j                  }t          j        	                    dd
d                              t          j
                  }t          ||d          \  }}t          |||g||gd           d S )Nr   r=   r$   rB   rC   r2   rD   r   r   rG   r   r   rI   Tr}   test_sce_mean_3d_log_probrN   rP   )r'   rZ   r$   rB   r8   r2   s         r9   +export_softmaxcrossentropy_mean_3d_log_probzCSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_3d_log_probV  s     	 {$$%:*%	 % 
 
 		qINN1a##**2:66Iaaf55<<RXFF -QEEEh 	q68$,		
 	
 	
 	
 	
 	
r;   c                 "   d} t           j                            dg ddg|           }t          j                            d           t          j                            dd                              t          j                  }t          j        	                    ddd	
                              t          j
                  }t          j        g dt          j                  }t          |||          }t          ||||g|gd           d S )Nr   r=   rd   rC   rD   r   r   rG   rH   rI   rf   r   )r&   test_sce_mean_weightrN   rj   rk   s         r9   'export_softmaxcrossentropy_mean_weightsz?SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weightss  s    	 {$$%"??E	 % 
 
 		qINN1a  ''
33""114"88??II(444BJGGG "!VG<<< 	vw'E'		
 	
 	
 	
 	
 	
r;   c                 .   d} t           j                            dg dddg|           }t          j                            d           t          j                            dd	                              t          j                  }t          j        	                    dd	d
                              t          j
                  }t          j        g dt          j                  }t          |||d          \  }}t          ||||g||gd           d S )Nr   r=   rd   rC   r2   rD   r   r   rG   rH   rI   rf   r   T)r&   r)   test_sce_mean_weight_log_probrN   rj   rq   s          r9   0export_softmaxcrossentropy_mean_weights_log_probzHSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weights_log_prob  s    	 {$$%"??*%	 % 
 
 		qINN1a  ''
33""114"88??II(444BJGGG -vgD
 
 
h
 	vw'8$0		
 	
 	
 	
 	
 	
r;   c                 |   d} t          j        d          }t          j                            dg ddg| |          }t           j                            d           t           j                            dd                              t           j	                  }t           j        
                    ddd	
                              t           j                  }t          j        d          |d<   t          j        g dt           j	                  }t          ||||          }t          ||||g|gd           d S )Nr   r   r=   rd   rC   rE   rF   r'   r(   r   rG   rH   rI   rf   r   r&   r(   test_sce_mean_weight_iirN   r   rX   rQ   rR   rS   rT   rU   rV   r   r    rW   r   r:   r   r'   r(   rZ   r$   r[   rl   r\   s          r9   *export_softmaxcrossentropy_mean_weights_iizBSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weights_ii  s.    	x{{ {$$%"??E% % 
 
 		qINN1a  ''
33""114"88??IIHQKKq	(444BJGGG "!VG,WWW 	vw'E*		
 	
 	
 	
 	
 	
r;   c                    d} t          j        d          }t          j                            dg dddg| |          }t           j                            d           t           j                            dd	                              t           j	                  }t           j        
                    dd	d
                              t           j                  }t          j        d          |d<   t          j        g dt           j	                  }t          ||||d          \  }}t          ||||g||gd           d S )Nr   r   r=   rd   rC   r2   r   r   rG   rH   rI   rf   r   Tr&   r(   r)    test_sce_mean_weight_ii_log_probrN   r   r'   r(   rZ   r$   r[   rl   r8   r2   s           r9   3export_softmaxcrossentropy_mean_weights_ii_log_probzKSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weights_ii_log_prob  sA    	x{{ {$$%"??*%% % 
 
 		qINN1a  ''
33""114"88??IIHQKKq	(444BJGGG -vgLt
 
 
h
 	vw'8$3		
 	
 	
 	
 	
 	
r;   c                 4   d} t          j        d          }t          j                            dddgdg| |          }t           j                            d           t           j                            d	d
                              t           j	                  }t           j        
                    dd
d                              t           j                  }t          j        d          |d<   t          |||          }t          |||g|gd           d S )Nr   r   r=   r$   rB   rC   r   r   r   rG   rH   rI   r(   test_sce_mean_no_weight_iirN   r   rX   rQ   rR   rS   rT   rU   rV   r   r    rW   r:   r   r'   r(   rZ   r$   r[   r\   s         r9   -export_softmaxcrossentropy_mean_no_weights_iizESoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_no_weights_ii  s
    	x{{ {$$%:E% % 
 
 		qINN1a  ''
33""114"88??IIHQKKq	 "!V,GGG 	!Vse:V	
 	
 	
 	
 	
 	
r;   c                 @   d} t          j        d          }t          j                            dddgddg| |          }t           j                            d	           t           j                            d
d                              t           j	                  }t           j        
                    d	dd                              t           j                  }t          j        d          |d	<   t          |||d          \  }}t          |||g||gd           d S )Nr   r   r=   r$   rB   rC   r2   r   r   r   rG   rH   rI   Tr(   r)   #test_sce_mean_no_weight_ii_log_probrN   r   r'   r(   rZ   r$   r[   r8   r2   s          r9   6export_softmaxcrossentropy_mean_no_weights_ii_log_probzNSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_no_weights_ii_log_prob  s    	x{{ {$$%:*%% % 
 
 		qINN1a  ''
33""114"88??IIHQKKq	 -vLt
 
 
h
 	v;8$6		
 	
 	
 	
 	
 	
r;   c                    d} t          j        d          }t          j                            dg ddg| |          }t           j                            d           t           j                            dd	d
                              t           j	                  }t           j        
                    dd	d                              t           j                  }t          j        d          |d         d<   t          j        g dt           j	                  }t          ||||          }t          ||||g|gd           d S )Nr   r   r=   rd   rC   r   r   r   rG   r   r   rI   g?g333333?g333333?g?g      ?r   r   test_sce_mean_weight_ii_3drN   r   r   s          r9   -export_softmaxcrossentropy_mean_weights_ii_3dzESoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weights_ii_3d4  s4    	x{{ {$$%"??E% % 
 
 		qINN1a##**2:66""116"::AA"(KKx{{q	!(444BJGGG "!VG,WWW 	vw'E-		
 	
 	
 	
 	
 	
r;   c                    d} t          j        d          }t          j                            dg dddg| |          }t           j                            d           t           j                            d	d
d                              t           j	                  }t           j        
                    dd
d                              t           j                  }t          j        d          |d         d<   t          j        g dt           j	                  }t          ||||d          \  }}t          ||||g||gd           d S )Nr   r   r=   rd   rC   r2   r   r   r   rG   r   r   rI   r   r   Tr   #test_sce_mean_weight_ii_3d_log_probrN   r   r   s           r9   6export_softmaxcrossentropy_mean_weights_ii_3d_log_probzNSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weights_ii_3d_log_probU  sG    	x{{ {$$%"??*%% % 
 
 		qINN1a##**2:66""116"::AA"(KKx{{q	!(444BJGGG -vgLt
 
 
h
 	vw'8$6		
 	
 	
 	
 	
 	
r;   c                 B   d} t          j        d          }t          j                            dddgdg| |          }t           j                            d           t           j                            d	d
d                              t           j	                  }t           j        
                    dd
d                              t           j                  }t          j        d          |d         d<   t          |||          }t          |||g|gd           d S )Nr   r   r=   r$   rB   rC   r   r   r   rG   r   rI   r   test_sce_mean_no_weight_ii_3drN   r   r   s         r9   0export_softmaxcrossentropy_mean_no_weights_ii_3dzHSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_no_weights_ii_3dx  s    	x{{ {$$%:E% % 
 
 		qINN1a##**2:66""116"::AA"(KKx{{q	! "!V,GGG 	v;E0		
 	
 	
 	
 	
 	
r;   c                 N   d} t          j        d          }t          j                            dddgddg| |          }t           j                            d	           t           j                            d
dd                              t           j	                  }t           j        
                    d	dd                              t           j                  }t          j        d          |d	         d	<   t          |||d          \  }}t          |||g||gd           d S )Nr   r   r=   r$   rB   rC   r2   r   r   r   rG   r   rI   Tr   &test_sce_mean_no_weight_ii_3d_log_probrN   r   r   s          r9   9export_softmaxcrossentropy_mean_no_weights_ii_3d_log_probzQSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_no_weights_ii_3d_log_prob  s%    	x{{ {$$%:*%% % 
 
 		qINN1a##**2:66""116"::AA"(KKx{{q	! -vLt
 
 
h
 	v;8$9		
 	
 	
 	
 	
 	
r;   c                    d} t          j        d          }t          j                            dg ddg| |          }t           j                            d           t           j                            dd	dd
                              t           j	                  }t           j        
                    dd	d                              t           j                  }t          j        d          |d         d         d<   t          j        g dt           j	                  }t          ||| ||          }t          ||||g|gd           d S )Nr   r   r=   rd   rC   r   r   r   rG      r   r   r   rI   r   r   )r'   r&   r(   test_sce_mean_weight_ii_4drN   r   r   s          r9   -export_softmaxcrossentropy_mean_weights_ii_4dzESoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weights_ii_4d  sC    	x{{ {$$%"??E% % 
 
 		qINN1aA&&--bj99""119"==DDRXNN(1++q	!Q(444BJGGG "v7
 
 

 	vw'E-		
 	
 	
 	
 	
 	
r;   c                    d} t          j        d          }t          j                            dg dddg| |          }t           j                            d           t           j                            d	d
dd                              t           j	                  }t           j        
                    dd
d                              t           j                  }t          j        d          |d         d         d<   t          j        g dt           j	                  }t          ||| ||d          \  }}t          ||||g||gd           d S )Nr   r   r=   rd   rC   r2   r   r   r   rG   r   r   rI   r   r   T)r'   r&   r(   r)   #test_sce_mean_weight_ii_4d_log_probrN   r   r   s           r9   6export_softmaxcrossentropy_mean_weights_ii_4d_log_probzNSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weights_ii_4d_log_prob  sT    	x{{ {$$%"??*%% % 
 
 		qINN1aA&&--bj99""119"==DDRXNN(1++q	!Q(444BJGGG -%
 
 
h 	vw'8$6		
 	
 	
 	
 	
 	
r;   c                 R   d} t          j        d          }t          j                            dddgdg| |          }t           j                            d           t           j                            d	d
dd                              t           j	                  }t           j        
                    dd
d                              t           j                  }t          j        d          |d         d         d<   t          ||| |          }t          |||g|gd           d S )Nr   r   r=   r$   rB   rC   r   r   r   rG   r   r   rI   r'   r(   test_sce_mean_no_weight_ii_4drN   r   r   s         r9   0export_softmaxcrossentropy_mean_no_weights_ii_4dzHSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_no_weights_ii_4d  s!    	x{{ {$$%:E% % 
 
 		qINN1aA&&--bj99""119"==DDRXNN(1++q	!Q "v
 
 

 	v;E0		
 	
 	
 	
 	
 	
r;   c                 ^   d} t          j        d          }t          j                            dddgddg| |          }t           j                            d	           t           j                            d
ddd                              t           j	                  }t           j        
                    d	dd                              t           j                  }t          j        d          |d	         d	         d	<   t          ||| |d          \  }}t          |||g||gd           d S )Nr   r   r=   r$   rB   rC   r2   r   r   r   rG   r   r   rI   Tr'   r(   r)   &test_sce_mean_no_weight_ii_4d_log_probrN   r   r   s          r9   9export_softmaxcrossentropy_mean_no_weights_ii_4d_log_probzQSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_no_weights_ii_4d_log_prob'  s.    	x{{ {$$%:*%% % 
 
 		qINN1aA&&--bj99""119"==DDRXNN(1++q	!Q -vTX
 
 
h
 	v;8$9		
 	
 	
 	
 	
 	
r;   c            
        d} t           j                            dg ddg|           }d\  }}}}}}}t          j                            d           t          j                            |||||||                              t          j                  }	t          j        	                    d|||||||f                              t          j
                  }
t          j                            |                              t          j                  }t          |	|
|| 	          }t          ||	|
|g|gd
           d S )Nr   r=   rd   rC   rD   r   rG      r   rG   r      r   rI   rh   !test_sce_NCd1d2d3d4d5_mean_weightrN   rP   )r'   rZ   r,   r-   dim1dim2dim3dim4dim5r$   r[   r&   r\   s                r9   .export_input_shape_is_NCd1d2d3d4d5_mean_weightzFSoftmaxCrossEntropyLoss.export_input_shape_is_NCd1d2d3d4d5_mean_weightI  sB   	{$$%"??E	 % 
 
 .A*1dD$d
	qINN1atT4>>EEbjQQ""AQdD$= # 
 

&

 	 ""))"*55!!VFiPPPvv&E4		
 	
 	
 	
 	
 	
r;   c            
        d} t           j                            dg dddg|           }d\  }}}}}}}t          j                            d           t          j                            |||||||                              t          j                  }	t          j        	                    d|||||||f	                              t          j
                  }
t          j                            |                              t          j                  }t          |	|
|| d
          \  }}t          ||	|
|g||gd           d S )Nr   r=   rd   rC   r2   rD   r   r   rI   Tro   *test_sce_NCd1d2d3d4d5_mean_weight_log_probrN   rP   )r'   rZ   r,   r-   r   r   r   r   r   r$   r[   r&   r8   r2   s                 r9   7export_input_shape_is_NCd1d2d3d4d5_mean_weight_log_probzOSoftmaxCrossEntropyLoss.export_input_shape_is_NCd1d2d3d4d5_mean_weight_log_probe  sW   	{$$%"??*%	 % 
 
 .A*1dD$d
	qINN1atT4>>EEbjQQ""AQdD$= # 
 

&

 	 ""))"*55,vf	
 
 
h 	vv&8$=		
 	
 	
 	
 	
 	
r;   c            
        d} t           j                            dddgdg|           }d\  }}}}}}}t          j                            d           t          j                            |||||||                              t          j                  }	t          j        	                    d|||||||f	                              t          j
                  }
t          |	|
| 
          }t          ||	|
g|gd           d S )NrA   r=   r$   rB   rC   rD   r   r   rI   rL   $test_sce_NCd1d2d3d4d5_none_no_weightrN   rP   )r'   rZ   r,   r-   r   r   r   r   r   r$   r[   r\   s               r9   1export_input_shape_is_NCd1d2d3d4d5_none_no_weightzISoftmaxCrossEntropyLoss.export_input_shape_is_NCd1d2d3d4d5_none_no_weight  s   	{$$%:E	 % 
 
 .A*1dD$d
	qINN1atT4>>EEbjQQ""AQdD$= # 
 

&

 	 "!VyAAAv;E7		
 	
 	
 	
 	
 	
r;   c            
        d} t           j                            dddgddg|           }d\  }}}}}}}t          j                            d	           t          j                            |||||||                              t          j                  }	t          j        	                    d	|||||||f
                              t          j
                  }
t          |	|
| d          \  }}t          ||	|
g||gd           d S )NrA   r=   r$   rB   rC   r2   rD   r   r   rI   Tr_   -test_sce_NCd1d2d3d4d5_none_no_weight_log_probrN   rP   )r'   rZ   r,   r-   r   r   r   r   r   r$   r[   r8   r2   s                r9   :export_input_shape_is_NCd1d2d3d4d5_none_no_weight_log_probzRSoftmaxCrossEntropyLoss.export_input_shape_is_NCd1d2d3d4d5_none_no_weight_log_prob  s/   	{$$%:*%	 % 
 
 .A*1dD$d
	qINN1atT4>>EEbjQQ""AQdD$= # 
 

&

 	 -v
 
 
h 	v;8$@		
 	
 	
 	
 	
 	
r;   c                    d} t          j        d          }t          j                            dg ddg| |          }d\  }}}t           j                            d           t           j                            |||                              t           j	                  }t           j        
                    d|||f	                              t           j                  }d|d         d<   t           j                            |                              t           j	                  }t          |||| |
          }	t          ||||g|	gd           d S )Nr   r   r=   rd   rC   r   r   rG   r   r   rI   r&   r'   r(   %test_sce_NCd1_mean_weight_negative_iirN   r   )
r'   r(   rZ   r,   r-   r   r$   r[   r&   r\   s
             r9   2export_input_shape_is_NCd1_mean_weight_negative_iizJSoftmaxCrossEntropyLoss.export_input_shape_is_NCd1_mean_weight_negative_ii  sD   	x||{$$%"??E% % 
 
 
1d
	qINN1a&&--bj99""11At9"==DDRXNNq	!""))"*55!vf	
 
 
 	vv&E8		
 	
 	
 	
 	
 	
r;   c                    d} t          j        d          }t          j                            dg dddg| |          }d\  }}}t           j                            d	           t           j                            |||                              t           j	                  }t           j        
                    d	|||f
                              t           j                  }d|d	         d	<   t           j                            |                              t           j	                  }t          |||| |d          \  }	}
t          ||||g|	|
gd           d S )Nr   r   r=   rd   rC   r2   r   r   r   rI   Tr&   r'   r(   r)   .test_sce_NCd1_mean_weight_negative_ii_log_probrN   r   )r'   r(   rZ   r,   r-   r   r$   r[   r&   r8   r2   s              r9   ;export_input_shape_is_NCd1_mean_weight_negative_ii_log_probzSSoftmaxCrossEntropyLoss.export_input_shape_is_NCd1_mean_weight_negative_ii_log_prob  sU   	x||{$$%"??*%% % 
 
 
1d
	qINN1a&&--bj99""11At9"==DDRXNNq	!""))"*55,%
 
 
h 	vv&8$A		
 	
 	
 	
 	
 	
r;   c                 T   d} t          j        d          }t          j                            dddgdg| |          }d\  }}}}}t           j                            d	           t           j                            |||||                              t           j	                  }t           j        
                    d	|||||f
                              t           j                  }	d|	d	         d	         d	         d	<   t          ||	| |          }
t          |||	g|
gd           d S )NrA   r=   r$   rB   rC   r   r   rG   r   r   rG   r   rI   r   ,test_sce_NCd1d2d3_none_no_weight_negative_iirN   r   )r'   r(   rZ   r,   r-   r   r   r   r$   r[   r\   s              r9   9export_input_shape_is_NCd1d2d3_none_no_weight_negative_iizQSoftmaxCrossEntropyLoss.export_input_shape_is_NCd1d2d3_none_no_weight_negative_ii  s:   	x||{$$%:E% % 
 
 "/1dD$
	qINN1atT2299"*EE""11AtT43H"IIPPH
 
  q	!Q!v
 
 
 	v;E?		
 	
 	
 	
 	
 	
r;   c                 `   d} t          j        d          }t          j                            dddgddg| |          }d	\  }}}}}t           j                            d
           t           j                            |||||                              t           j	                  }t           j        
                    d
|||||f                              t           j                  }	d|	d
         d
         d
         d
<   t          ||	| |d          \  }
}t          |||	g|
|gd           d S )NrA   r   r=   r$   rB   rC   r2   r   r   r   rI   Tr   5test_sce_NCd1d2d3_none_no_weight_negative_ii_log_probrN   r   )r'   r(   rZ   r,   r-   r   r   r   r$   r[   r8   r2   s               r9   Bexport_input_shape_is_NCd1d2d3_none_no_weight_negative_ii_log_probzZSoftmaxCrossEntropyLoss.export_input_shape_is_NCd1d2d3_none_no_weight_negative_ii_log_prob  sG   	x||{$$%:*%% % 
 
 "/1dD$
	qINN1atT2299"*EE""11AtT43H"IIPPH
 
  q	!Q,vTX
 
 
h 	v;8$H		
 	
 	
 	
 	
 	
r;   c                    d} t          j        d          }t          j                            dg ddg| |          }d\  }}t           j                            d           t           j                            ||                              t           j	                  }t           j        
                    d||	                              t           j                  }d|d<   t           j                            |                              t           j	                  }t          |||| |
          }t          ||||g|gd           d S )Nr   
   r=   rd   rC   r   r   rG   r   rI   r   $test_sce_NCd1d2d3_sum_weight_high_iirN   r   )	r'   r(   rZ   r,   r-   r$   r[   r&   r\   s	            r9   1export_input_shape_is_NCd1d2d3_sum_weight_high_iizISoftmaxCrossEntropyLoss.export_input_shape_is_NCd1d2d3_sum_weight_high_ii>  s8   	x||{$$%"??E% % 
 
 1
	qINN1a  ''
33""11A"77>>rxHHq	""))"*55!vf	
 
 
 	vv&E7		
 	
 	
 	
 	
 	
r;   c                    d} t          j        d          }t          j                            dg dddg| |          }d\  }}t           j                            d	           t           j                            ||                              t           j	                  }t           j        
                    d	||
                              t           j                  }d|d	<   t           j                            |                              t           j	                  }t          |||| |d          \  }}	t          ||||g||	gd           d S )Nr   r   r=   rd   rC   r2   r   r   r   rI   Tr   -test_sce_NCd1d2d3_sum_weight_high_ii_log_probrN   r   )
r'   r(   rZ   r,   r-   r$   r[   r&   r8   r2   s
             r9   :export_input_shape_is_NCd1d2d3_sum_weight_high_ii_log_probzRSoftmaxCrossEntropyLoss.export_input_shape_is_NCd1d2d3_sum_weight_high_ii_log_prob]  sI   	x||{$$%"??*%% % 
 
 1
	qINN1a  ''
33""11A"77>>rxHHq	""))"*55,%
 
 
h 	vv&8$@		
 	
 	
 	
 	
 	
r;   N)r>   r?   )&__name__
__module____qualname__staticmethodr]   rb   rm   rr   ru   rx   r{   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r;   r9   r=   r=   \   sm       N N N \N. 
 
 
 \
< 
 
 
 \
: 
 
 
 \
> M M M \M. 
 
 
 \
< N N N \N. 
 
 
 \
8 L L L \L. 
 
 
 \
8 
 
 
 \
: 
 
 
 \
> 
 
 
 \
@  
  
  
 \ 
D 
 
 
 \
8 
 
 
 \
B 
 
 
 \
@  
  
  
 \ 
D 
 
 
 \
> 
 
 
 \
B  
  
  
 \ 
D %
 %
 %
 \%
N 
 
 
 \
B 
 
 
 \
B 
 
 
 \
6 
 
 
 \
: 
 
 
 \
4 
 
 
 \
8 
 
 
 \
< !
 !
 !
 \!
F 
 
 
 \
> 
 
 
 \
> 
 
 
 \
< !
 !
 !
 \!
 !
 !
r;   r=   )Nr   NN)
__future__r   numpyr   rQ   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r:   r=   r   r;   r9   <module>r      s    # " " " " "      , , , , , , . . . . . . OSL L L L^c
 c
 c
 c
 c
d c
 c
 c
 c
 c
r;   