Di ddlZddlZddlZdZdZdZdZ ddededed e d e d e f d Z d Z dZ dZdejfdZdZdZdZdefdZdZdZdZdZdZdZy)Nc"d}|D]}||z} |S)z6Compute the product of all elements in the input list.)in_listres_s W/var/www/pod-logistic/pod-ai/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_sizegroupsbias transposec|r%|d}t|||zzt|ddzS|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!r"out_cin_cs r calculate_conv2d_flopsr' sa Aj!Uf_5{128OOO!}k"dfn5{128OOOr 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_normr04s   q:~. //r cy)z[Calculates the FLOPs for a ReLU activation function based on the input tensor's dimensions.rrr/s r calculate_relu_flopsr29s 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_relur5>s' MM01   s:/ 00r ch|}|dz }|}|||z|zz}tjt|gS)zJCompute FLOPs for a softmax activation given batch size and feature count.rrrr4) batch_size nfeatures total_exp total_add total_div total_opss r calculate_softmaxr>DsAIA 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_matmulrWls6*%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_mulrYs r c|S)z\Computes the total scalar multiplications required for power operations based on input size.rr/s r counter_powr\xrZr c|S)ziCalculate the total number of scalar operations required for a square root operation given an input size.rr/s r counter_sqrtr^}rZr c|S)z]Calculate the total number of scalar operations for a division operation given an input size.rr/s r counter_divr`rZr )FF)r)numpyrTrr rrrlistr4boolr'r-r0r2Tensorr5r>r@rCstrrMrQrWrYr\r^r`rr r rfs  G # pu P P#' P6: PDG POS Phl PY 0  1u||1 01 > 0S 0@ 1   r