我想在TensorFlow中编写RMS池层。这很像，tf.nn.avg_pool()但与其计算简单平均值，不如计算均方根平均值。
我想最初使用numpy编写此代码；我不在乎它在GPU上运行（但是稍后我想编写一个GPU版本）
尝试
tf.sqrt(tf.nn.avg_pool(tf.square(value - \
tf.reduce_mean(value, reduction_indices=[0, 1, 2])),
<ksize>, <strides>, <padding>))
请注意，l_2池是常用的，而不是RMS池。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句