Как рассчитать операции с плавающей запятой для рекуррентной нейронной сети (RNN). Архитектура слоя - это Gated Recurrent Unit GRU и полностью связанный слой.
Я использовал https://github.com/Lyken17/pytorch-OpCounter, чтобы найти число флопов, но я пытаюсь вычислить математические вычисления вручную
self.rnn_gru = nn.GRU(input_size=1, hidden_size=20, num_layers=1, batch_first=True, dropout=0.2)
self.linear = nn.Linear(in_features=20, out_features=4)