a óÃÃfÃã@sÒddlZddlZddlZdd„Zdd„Zdd„Zdd „Zd,eeee e d œd d „Z dd„Z dd„Z dd„Zejdœdd„Zdd„Zdd„Zdd„Zedœdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„ZdS)-éNcCsd}|D] }||9}q|S)z6Compute the product of all elements in the input list.é©)Úin_listÚresÚ_rrúQ/var/www/html/django/DPS/env/lib/python3.9/site-packages/thop/vision/calc_func.pyÚl_prods rcCst|ƒS)z6Calculate the sum of all numerical elements in a list.©Úsum)rrrrÚl_sumsr cCstdd„|DƒƒS)z`Calculate the total number of parameters in a list of tensors using the product of their shapes.css|]}t | ¡g¡VqdS)N)ÚtorchÚ DoubleTensorZnelement)Ú.0ÚprrrÚ óz'calculate_parameters..r )Z param_listrrrÚcalculate_parameterssrcCs t dg¡S)z?Initializes and returns a tensor with all elements set to zero.r©r r rrrrÚcalculate_zero_opssrF)Ú input_sizeÚ output_sizeÚ kernel_sizeÚgroupsÚbiascCs,|d}|}t|ƒ||t|dd…ƒS)zdCalculate FLOPs for a Conv2D layer using input/output sizes, kernel size, groups, and the bias flag.réN)r)rrrrrZin_cÚgrrrÚcalculate_conv2d_flopssrcCs&t d¡t |||||g¡S)zgCalculate FLOPs for convolutional layers given bias, kernel size, output size, in_channels, and groups.zThis API is being deprecated.)ÚwarningsÚwarnr r )rrrZ in_channelÚgrouprrrÚcalculate_conv's r cCst d|g¡S)zACompute the L2 norm of a tensor or array based on its input size.rr©rrrrÚcalculate_norm-sr"cCsdS)z[Calculates the FLOPs for a ReLU activation function based on the input tensor's dimensions.rrr!rrrÚcalculate_relu_flops2sr#r!cCst d¡t t|ƒg¡S)zKConvert an input tensor to a DoubleTensor with the same value (deprecated).zThis API is being deprecated)rrr r Úintr!rrrÚcalculate_relu7s r%cCs0|}|d}|}||||}t t|ƒg¡S)zJCompute FLOPs for a softmax activation given batch size and feature count.r©r r r$)Z batch_sizeZ nfeaturesZ total_expZ total_addÚ total_divÚ total_opsrrrÚcalculate_softmax=s r)cCst t|ƒg¡S)zs*