
    Ng                     R    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 )	    N)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                        k/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/backend/test/case/node/softmaxcrossentropy.pysoftmaxcrossentropyr9      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~%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returnNc                     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rC   rD   nameonnxhelper	make_noder   randomseedrandr   r   randintint64r9   r   r&   noder#   labelssces        r8   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 )Nr?   r<   r#   r@   rA   r1   rB   r   r   rE   rF   rG   Tr&   r(   test_sce_none_log_probrL   rN   r&   rX   r#   rY   r7   r1   s         r8   (export_softmaxcrossentropy_none_log_probz@SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_none_log_probs   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 )Nr?   r<   r#   r@   wrA   rB   r   r   rE   rF   rG   ?gffffff?g?re   re   r
   r%   r&   test_sce_none_weightsrL   rO   rP   rQ   r   rR   rS   rT   r   r   rU   rV   r   r9   r   r&   rX   r#   rY   weightsrZ   s         r8   '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 )Nr?   r<   rb   rA   r1   rB   r   r   rE   rF   rG   rd   r
   Tr%   r&   r(   test_sce_none_weights_log_probrL   rh   r&   rX   r#   rY   rj   r7   r1   s          r8   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#   r@   rA   rB   r   r   rE   rF   rG   rJ   test_sce_sumrL   rN   rW   s        r8   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#   r@   rA   r1   rB   r   r   rE   rF   rG   Tr]   test_sce_sum_log_probrL   rN   r_   s         r8   '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#   r@   rA   rB   r   r   rE   rF   rG   test_sce_meanrL   rN   rW   s        r8   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#   r@   rA   r1   rB   r   r   rE   rF   rG   Tr(   test_sce_mean_log_probrL   rN   r_   s         r8   (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#   r@   rA   rB   r   r   rE   r   r   r   rG   test_sce_mean_3drL   rN   )r&   rX   r#   r@   rZ   s        r8   "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#   r@   rA   r1   rB   r   r   rE   r   r   rG   Tr{   test_sce_mean_3d_log_probrL   rN   )r&   rX   r#   r@   r7   r1   s         r8   +export_softmaxcrossentropy_mean_3d_log_probzCSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_3d_log_probT  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<   rb   rA   rB   r   r   rE   rF   rG   rd   r
   )r%   test_sce_mean_weightrL   rh   ri   s         r8   'export_softmaxcrossentropy_mean_weightsz?SoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weightsq  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<   rb   rA   r1   rB   r   r   rE   rF   rG   rd   r
   T)r%   r(   test_sce_mean_weight_log_probrL   rh   ro   s          r8   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<   rb   rA   rC   rD   r&   r'   r   rE   rF   rG   rd   r
   r%   r'   test_sce_mean_weight_iirL   r   rV   rO   rP   rQ   rR   rS   rT   r   r   rU   r   r9   r   r&   r'   rX   r#   rY   rj   rZ   s          r8   *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<   rb   rA   r1   r   r   rE   rF   rG   rd   r
   Tr%   r'   r(    test_sce_mean_weight_ii_log_probrL   r   r&   r'   rX   r#   rY   rj   r7   r1   s           r8   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#   r@   rA   r   r   r   rE   rF   rG   r'   test_sce_mean_no_weight_iirL   r   rV   rO   rP   rQ   rR   rS   rT   r   r   rU   r9   r   r&   r'   rX   r#   rY   rZ   s         r8   -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#   r@   rA   r1   r   r   r   rE   rF   rG   Tr'   r(   #test_sce_mean_no_weight_ii_log_probrL   r   r&   r'   rX   r#   rY   r7   r1   s          r8   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<   rb   rA   r   r   r   rE   r   r   rG   g?g333333?g333333?g?g      ?r
   r   test_sce_mean_weight_ii_3drL   r   r   s          r8   -export_softmaxcrossentropy_mean_weights_ii_3dzESoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weights_ii_3d2  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<   rb   rA   r1   r   r   r   rE   r   r   rG   r   r
   Tr   #test_sce_mean_weight_ii_3d_log_probrL   r   r   s           r8   6export_softmaxcrossentropy_mean_weights_ii_3d_log_probzNSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_weights_ii_3d_log_probS  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#   r@   rA   r   r   r   rE   r   rG   r   test_sce_mean_no_weight_ii_3drL   r   r   s         r8   0export_softmaxcrossentropy_mean_no_weights_ii_3dzHSoftmaxCrossEntropyLoss.export_softmaxcrossentropy_mean_no_weights_ii_3dv  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#   r@   rA   r1   r   r   r   rE   r   rG   Tr   &test_sce_mean_no_weight_ii_3d_log_probrL   r   r   s          r8   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<   rb   rA   r   r   r   rE      r   r   r   rG   r   r
   )r&   r%   r'   test_sce_mean_weight_ii_4drL   r   r   s          r8   -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<   rb   rA   r1   r   r   r   rE   r   r   rG   r   r
   T)r&   r%   r'   r(   #test_sce_mean_weight_ii_4d_log_probrL   r   r   s           r8   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#   r@   rA   r   r   r   rE   r   r   rG   r&   r'   test_sce_mean_no_weight_ii_4drL   r   r   s         r8   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#   r@   rA   r1   r   r   r   rE   r   r   rG   Tr&   r'   r(   &test_sce_mean_no_weight_ii_4d_log_probrL   r   r   s          r8   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<   rb   rA   rB   r   rE      r   rE   r      r   rG   rf   !test_sce_NCd1d2d3d4d5_mean_weightrL   rN   )r&   rX   r+   r,   dim1dim2dim3dim4dim5r#   rY   r%   rZ   s                r8   .export_input_shape_is_NCd1d2d3d4d5_mean_weightzFSoftmaxCrossEntropyLoss.export_input_shape_is_NCd1d2d3d4d5_mean_weightG  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<   rb   rA   r1   rB   r   r   rG   Trm   *test_sce_NCd1d2d3d4d5_mean_weight_log_probrL   rN   )r&   rX   r+   r,   r   r   r   r   r   r#   rY   r%   r7   r1   s                 r8   7export_input_shape_is_NCd1d2d3d4d5_mean_weight_log_probzOSoftmaxCrossEntropyLoss.export_input_shape_is_NCd1d2d3d4d5_mean_weight_log_probc  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 )Nr?   r<   r#   r@   rA   rB   r   r   rG   rJ   $test_sce_NCd1d2d3d4d5_none_no_weightrL   rN   )r&   rX   r+   r,   r   r   r   r   r   r#   rY   rZ   s               r8   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 )Nr?   r<   r#   r@   rA   r1   rB   r   r   rG   Tr]   -test_sce_NCd1d2d3d4d5_none_no_weight_log_probrL   rN   )r&   rX   r+   r,   r   r   r   r   r   r#   rY   r7   r1   s                r8   :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<   rb   rA   r   r   rE   r   r   rG   r%   r&   r'   %test_sce_NCd1_mean_weight_negative_iirL   r   )
r&   r'   rX   r+   r,   r   r#   rY   r%   rZ   s
             r8   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<   rb   rA   r1   r   r   r   rG   Tr%   r&   r'   r(   .test_sce_NCd1_mean_weight_negative_ii_log_probrL   r   )r&   r'   rX   r+   r,   r   r#   rY   r%   r7   r1   s              r8   ;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 )Nr?   r<   r#   r@   rA   r   r   rE   r   r   rE   r   rG   r   ,test_sce_NCd1d2d3_none_no_weight_negative_iirL   r   )r&   r'   rX   r+   r,   r   r   r   r#   rY   rZ   s              r8   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 )Nr?   r   r<   r#   r@   rA   r1   r   r   r   rG   Tr   5test_sce_NCd1d2d3_none_no_weight_negative_ii_log_probrL   r   )r&   r'   rX   r+   r,   r   r   r   r#   rY   r7   r1   s               r8   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<   rb   rA   r   r   rE   r   rG   r   $test_sce_NCd1d2d3_sum_weight_high_iirL   r   )	r&   r'   rX   r+   r,   r#   rY   r%   rZ   s	            r8   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<   rb   rA   r1   r   r   r   rG   Tr   -test_sce_NCd1d2d3_sum_weight_high_ii_log_probrL   r   )
r&   r'   rX   r+   r,   r#   rY   r%   r7   r1   s
             r8   :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:   )r=   N)&__name__
__module____qualname__staticmethodr[   r`   rk   rp   rs   rv   ry   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:   r8   r<   r<   Z   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)	numpyr   rO   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r9   r<   r   r:   r8   <module>r      s   
      , , , , , , . . . . . .K K K K\c
 c
 c
 c
 c
d c
 c
 c
 c
 c
r:   