В настоящее время я использую существующую реализацию Keras определенной модели, и я хотел бы изучить влияние различных реализаций умножения на скорость и точность вычислений.
Есть ли простой способ заменить умножение Keras (TensorFlow), которое используется в его слоях Dense и Conv (и других уже существующих), на пользовательское?
Идея также состоит в том, чтобы увидеть разницу между обучением с обычным умножением + тестированием с пользовательским умножением и выполнением обоих с пользовательским умножением.
Итак, я ищу решение, похожее на:
import tensorflow as tf
tf.__mul__ = custom_mult
и заменит все операции умножения в слоях Keras по умолчанию моей собственной реализацией.