j ddlZddlZddlZdZdZdZdZ ddeded ed e d e d e f d Z dZ dZ dZdejfdZdZdZdZdefdZdZdZdZdZdZdZdS)Ncd}|D]}||z}|S)z6Compute the product of all elements in the input list.)in_listres_s Z/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/thop/vision/calc_func.pyl_prodr s& C  q Jc t|S)z6Calculate the sum of all numerical elements in a list.sum)rs r l_sumrs w<<r c4td|DS)z`Calculate the total number of parameters in a list of tensors using the product of their shapes.c3dK|]+}tj|gV,dS)N)torch DoubleTensornelement).0ps r z'calculate_parameters..s7FFau!1::<<.11FFFFFFr r ) param_lists r calculate_parametersrs FF:FFF F FFr c,tjdgS)z?Initializes and returns a tensor with all elements set to zero.rrrrr r calculate_zero_opsrs  qc " ""r F input_size output_size kernel_sizegroupsbias transposec|r5|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' sw PAj!!Uf_5{1228O8OOO!}k""dfn5{1228O8OOOr cltjdtj|||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-.s< M1222  {j5.@;.NQU.UVW X XXr c2tjd|zgS)zACompute the L2 norm of a tensor or array based on its input size.r$rrs r calculate_normr04s  q:~. / //r cdS)z[Calculates the FLOPs for a ReLU activation function based on the input tensor's dimensions.rrr/s r calculate_relu_flopsr29s 1r cntjdtjt |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- M0111  s:/ 0 00r cn|}|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>DsCIA IIi)3i?@I  s9~~. / //r cFtjt|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_matmulrWls9*%%J(;''K 7:  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  GGG ### pu P P P#' P6: PDG POS Phl P P P PYYY 000    1u|1111 000111 >>> 0S 0 0 0 0@@@ 111   r