Как написать новый слой пула в TensorFlow?

python neural-network tensorflow

467 просмотра

1 ответ

Я хотел бы написать слой пула RMS в TensorFlow. Это похоже на то, tf.nn.avg_pool()но вместо простого среднего следует рассчитать среднеквадратичное значение.

Я хотел бы написать это, используя numpy изначально; Меня не волнует, работает ли он на GPU (но позже я бы хотел написать версию для GPU)

Автор: Alex I Источник Размещён: 08.11.2019 11:07

Ответы (1)


1 плюс

Решение

Пытаться

tf.sqrt(tf.nn.avg_pool(tf.square(value - \
     tf.reduce_mean(value, reduction_indices=[0, 1, 2])),
     <ksize>, <strides>, <padding>))

Обратите внимание, что пулы l_2 обычно используются, а не RMS.

Автор: TFUser Размещён: 21.08.2016 03:03
Вопросы из категории :
32x32