torch.utils.flop_counter.conv_flop_count

torch.utils.flop_counter.conv_flop_count(x_shape, w_shape, out_shape, transposed=False) [исходный код]

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

Обратите внимание: учитываются только операции умножения. Вычисления для смещения не учитываются. Число операций для транспонированной свёртки вычисляется по формуле flops = (x_shape[2:] * prod(w_shape) * batch_size). :param x_shape: Форма входных данных до свёртки. :type x_shape: list(int) :param w_shape: Форма фильтра. :type w_shape: list(int) :param out_shape: Форма выходных данных после свёртки. :type out_shape: list(int) :param transposed: является ли свёртка транспонированной :type transposed: bool

Возвращает:

число операций

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

int