torch.utils.cpp_extension

Определение

Инструмент PyTorch для компиляции и загрузки кастомных C++ и CUDA расширений, используемый для интеграции оптимизированных операторов.

Где встречается

• 53. Реализовать selective scan (Mamba)

Навигация

• Индекс терминов
• Индекс разборов
• Оглавление