h SSKrSSKrSSKrSrSrSrSrSS\S\S\S \ S \ S \ 4 S jjr S r Sr SrS\R4SjrSrSrSrS\4SjrSrSrSrSrSrSrg)Nc$SnUHnX-nM U$)z6Compute the product of all elements in the input list.)in_listres_s O/home/james-whalen/.local/lib/python3.13/site-packages/thop/vision/calc_func.pyl_prodr s C   Jc[U5$)z6Calculate the sum of all numerical elements in a list.sum)rs r l_sumrs w<r c&[SU55$)z`Calculate the total number of parameters in a list of tensors using the product of their shapes.c3l# UH*n[R"UR5/5v M, g7f)N)torch DoubleTensornelement).0ps r 'calculate_parameters..s&F:au!!1::<.11:s24r ) param_lists r calculate_parametersrs F:F FFr c0[R"S/5$)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 transposecU(a$USn[U5Xc--[USS5-$USn[U5Xs--[USS5-$)zdCalculate FLOPs for a Conv2D layer using input/output sizes, kernel size, groups, and the bias flag.rN)r )rrr r!r"r#out_cin_cs r calculate_conv2d_flopsr( s] Aj!U_5{128OOO!}k"dn5{128OOOr cr[R"S5 [R"X#U- U-U--/5$)zgCalculate FLOPs for convolutional layers given bias, kernel size, output size, in_channels, and groups.zThis API is being deprecated.)warningswarnrr)r"r r in_channelgroups r calculate_convr..s6 MM12   {5.@;.NQU.UVW XXr c6[R"SU-/5$)zACompute the L2 norm of a tensor or array based on its input size.r%rrs r calculate_normr14s   q:~. //r cg)z[Calculates the FLOPs for a ReLU activation function based on the input tensor's dimensions.rrr0s r calculate_relu_flopsr39s r cn[R"S5 [R"[ U5/5$)zKConvert an input tensor to a DoubleTensor with the same value (deprecated).zThis API is being deprecated)r*r+rrintr0s r calculate_relur6>s' MM01   s:/ 00r chUnUS- nUnXU-U--n[R"[U5/5$)zJCompute FLOPs for a softmax activation given batch size and feature count.rrrr5) batch_size nfeatures total_exp total_add total_div total_opss r calculate_softmaxr?Ds?IA II)3i?@I   s9~. //r cB[R"[U5/5$)zs r calculate_upsamplerNYseI yX  R  Q  Z   s9~. //r cF[R"[X-5/5$)zTCalculate the linear operation count for given input feature and number of elements.r8) in_feature num_elementss r calculate_linearrRgs   s:#<=> ??r c[R"U5n[R"U5n[R"U5US-$)z`Calculate the total number of operations for matrix multiplication given input and output sizes.)nparrayprod)rrs r counter_matmulrXls6*%J((;'K 77: R 00r cU$)z^Calculate the total number of operations for element-wise multiplication given the input size.rr0s r counter_mulrZs r cU$)z\Computes the total scalar multiplications required for power operations based on input size.rr0s r counter_powr]xr[r cU$)ziCalculate the total number of scalar operations required for a square root operation given an input size.rr0s r counter_sqrtr_}r[r cU$)z]Calculate the total number of scalar operations for a division operation given an input size.rr0s r counter_divrar[r )FF)r*numpyrUrr rrrlistr5boolr(r.r1r3Tensorr6r?rArDstrrNrRrXrZr]r_rarr r rgs  G # pu P P#' P6: PDG POS Phl PY 0  1u||1 01 > 0S 0@ 1   r