o  h  @sddlZddlZddlZddZddZddZdd Zd/d ed ed ede de f ddZ ddZ ddZ ddZd ejfddZddZddZddZd efd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.ZdS)0NcCsd}|D]}||9}q|S)z6Compute the product of all elements in the input list.)in_listres_rri/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/thop/vision/calc_func.pyl_prod s rcCst|S)z6Calculate the sum of all numerical elements in a list.sum)rrrrl_sumsr cCstdd|DS)z`Calculate the total number of parameters in a list of tensors using the product of their shapes.css |] }t|gVqdS)N)torch DoubleTensornelement).0prrr sz'calculate_parameters..r ) param_listrrrcalculate_parameterssrcCs tdgS)z?Initializes and returns a tensor with all elements set to zero.rr r rrrrcalculate_zero_opss rF input_size output_size kernel_sizegroupsbiascCs,|d}|}t|||t|ddS)zdCalculate FLOPs for a Conv2D layer using input/output sizes, kernel size, groups, and the bias flag.rN)r)rrrrrin_cgrrrcalculate_conv2d_flops s rcCs&tdt|||||gS)zgCalculate FLOPs for convolutional layers given bias, kernel size, output size, in_channels, and groups.zThis API is being deprecated.)warningswarnr r )rrr in_channelgrouprrrcalculate_conv)s r#cCstd|gS)zACompute the L2 norm of a tensor or array based on its input size.rrrrrrcalculate_norm/r%cCsdS)z[Calculates the FLOPs for a ReLU activation function based on the input tensor's dimensions.rrr$rrrcalculate_relu_flops4r'cCstdtt|gS)zKConvert an input tensor to a DoubleTensor with the same value (deprecated).zThis API is being deprecated)rr r r intr$rrrcalculate_relu9s r*cCs0|}|d}|}||||}tt|gS)zJCompute FLOPs for a softmax activation given batch size and feature count.rr r r)) batch_size nfeatures total_exp total_add total_div total_opsrrrcalculate_softmax?s r2cCstt|gS)zcCstt||gS)zTCalculate the linear operation count for given input feature and number of elements.r+) in_feature num_elementsrrrcalculate_linearbsrAcCs&t|}t|}t||dS)z`Calculate the total number of operations for matrix multiplication given input and output sizes.)nparrayprod)rrrrrcounter_matmulgs  rFcC|S)z^Calculate the total number of operations for element-wise multiplication given the input size.rr$rrr counter_mulnr(rHcCrG)z\Computes the total scalar multiplications required for power operations based on input size.rr$rrr counter_powsr(rIcCrG)ziCalculate the total number of scalar operations required for a square root operation given an input size.rr$rrr counter_sqrtxr(rJcCrG)z]Calculate the total number of scalar operations for a division operation given an input size.rr$rrr counter_div}r(rK)F)rnumpyrCr rr rrlistr)boolrr#r%r'Tensorr*r2r3r5strr>rArFrHrIrJrKrrrrs,