Rate this Page

torch.utils.flop_counter.conv_flop#

torch.utils.flop_counter.conv_flop(x_shape, w_shape, _bias, _stride, _padding, _dilation, transposed, *args, out_shape=None, **kwargs)[source]#

Count flops for convolution.

Return type:

int