torch.utils.flop_counter.baddbmm_flop

torch.utils.flop_counter.baddbmm_flop(self_shape, a_shape, b_shape, *args, **kwargs)

Count flops for the baddbmm operation.

Return type:

int