torch.utils.flop_counter.baddbmm_flop

torch.utils.flop_counter.baddbmm_flop(self_shape, a_shape, b_shape, *args, **kwargs) [исходный код]

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

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

int