torch.nn.attention.current_flash_attention_impl

torch.nn.attention.current_flash_attention_impl() [исходный код]

Возвращает имя текущей активированной реализации flash attention, если она есть.

None означает, что пользовательская реализация не активирована.

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

str | None