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]

Подсчитывает количество операций с плавающей запятой для свёртки.

Тип возвращаемого значения:

int