在现有的keras(tensorflow)模型中替换乘法运算符

时间:2019-05-24 07:32:38

标签: tensorflow keras matrix-multiplication multiplication

我目前正在使用某个模型的现有Keras实现,并且我想研究不同乘法实现对其计算速度和准确性的影响。

是否有一种简单的方法可以用自定义层替换在其Dense和Conv(以及其他先前存在的)层中使用的Keras(TensorFlow)乘法?

该想法还旨在了解普通乘法训练与自定义乘法测试与自定义乘法两者之间的区别。

所以我正在寻找一种类似的解决方案:

import tensorflow as tf
tf.__mul__ = custom_mult

,并将用我自己的实现替换Keras默认层中的所有乘法运算。

0 个答案:

没有答案