torch.utils.flop_counter.bmm_flop

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

Count flops for the bmm operation.

Return type:

int