как изменить рН клетки в pytorch?

Question

Если я хочу изменить правила вычислений в ячейке RNN (например, ячейке GRU), что мне делать?
Я не хочу реализовывать это с помощью цикла for или while, учитывая проблему эффективности.
Я просмотрел исходный код pytorch, но кажется, что основные компоненты ячеек rnn реализованы в коде c, который я не могу найти и изменить.Вы можете ответить на этот вопрос на примере: внедрить ячейку GRU без существующей версии.

спасибо ~

Answer 1

Да, вы реализуете это через цикл for или while.Начиная с Pytorch 1.0 есть JIT https://pytorch.org/docs/stable/jit.html, который работает довольно хорошо (вероятно, лучше использовать последнюю версию git PyTorch из-за недавних улучшений JIT), и в зависимости от вашей сети и реализации может работать так же быстро, как и собственный PyTorchРеализация C ++ (но все же медленнее, чем CuDNN).

Примеры реализации можно посмотреть на https://github.com/pytorch/benchmark/blob/master/rnns/fastrnns/custom_lstms.py