torch.utils.flop_counter.baddbmm_flop

torch.utils.flop_counter.baddbmm_flop(self_shape, a_shape, b_shape, out_shape=None, **kwargs)

Count flops for the baddbmm operation.

Return type:

int