|h  ddlZddlZddlZdZdZdZdZddededed e d e f d Z d Z d Z dZdejfdZdZdZdZdefdZdZdZdZdZdZdZy)Nc"d}|D]}||z} |S)z6Compute the product of all elements in the input list.)in_listres_s T/var/www/html/test/engine/venv/lib/python3.12/site-packages/thop/vision/calc_func.pyl_prodr s$ C  q Jct|S)z6Calculate the sum of all numerical elements in a list.sum)rs r l_sumrs w<r c&td|DS)z`Calculate the total number of parameters in a list of tensors using the product of their shapes.c3dK|](}tj|jg*yw)N)torch DoubleTensornelement).0ps r z'calculate_parameters..s#Fau!!1::<.1Fs.0r ) param_lists r calculate_parametersrs F:F FFr c.tjdgS)z?Initializes and returns a tensor with all elements set to zero.rrrrr r calculate_zero_opsrs   qc ""r input_size output_size kernel_sizegroupsbiascP|d}|}t|||zzt|ddzS)zdCalculate FLOPs for a Conv2D layer using input/output sizes, kernel size, groups, and the bias flag.rN)r )rrrr r!in_cgs r calculate_conv2d_flopsr& s6 a=DA + $!) ,vk!"o/F FFr cptjdtj|||z |z|zzgS)zgCalculate FLOPs for convolutional layers given bias, kernel size, output size, in_channels, and groups.zThis API is being deprecated.)warningswarnrr)r!rr in_channelgroups r calculate_convr,)s8 MM12   {j5.@;.NQU.UVW XXr c4tjd|zgS)zACompute the L2 norm of a tensor or array based on its input size.r#rrs r calculate_normr//s   q:~. //r cy)z[Calculates the FLOPs for a ReLU activation function based on the input tensor's dimensions.rrr.s r calculate_relu_flopsr14s r cjtjdtjt |gS)zKConvert an input tensor to a DoubleTensor with the same value (deprecated).zThis API is being deprecated)r(r)rrintr.s r calculate_relur49s' MM01   s:/ 00r ch|}|dz }|}|||z|zz}tjt|gS)zJCompute FLOPs for a softmax activation given batch size and feature count.rrrr3) batch_size nfeatures total_exp total_add total_div total_opss r calculate_softmaxr=?sAIA IIi)3i?@I   s9~. //r c@tjt|gS)z ??r ctj|}tj|}tj||dzS)z`Calculate the total number of operations for matrix multiplication given input and output sizes.)nparrayprod)rrs r counter_matmulrVgs6*%J((;'K 77: R 00r c|S)z^Calculate the total number of operations for element-wise multiplication given the input size.rr.s r counter_mulrXn r c|S)z\Computes the total scalar multiplications required for power operations based on input size.rr.s r counter_powr[srYr c|S)ziCalculate the total number of scalar operations required for a square root operation given an input size.rr.s r counter_sqrtr]xrYr c|S)z]Calculate the total number of scalar operations for a division operation given an input size.rr.s r counter_divr_}rYr )F)r(numpyrSrr rrrlistr3boolr&r,r/r1Tensorr4r=r?rBstrrLrPrVrXr[r]r_rr r res  G # GtG$GTG[^GfjGY 0  1u||1 01 > 0S 0@ 1   r