torch.utils.flop_counter.sdpa_flop

torch.utils.flop_counter.sdpa_flop(query_shape, key_shape, value_shape, *args, out_shape=None, **kwargs) [исходный код]

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

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

int