Mãos à Obra: Aprendizado de Máquina com Scikit-Learn e TensorFlow

Compartilhando Variáveis | 253

Compartilhando Variáveis

Se você quiser compartilhar uma variável entre vários componentes do seu grafo, uma op-

ção simples é criá-la primeiro e depois passá-la como um parâmetro para as funções que a

necessitam. Por exemplo, suponha que você deseja controlar o limiar da ReLU (atualmente

codificado para 0) utilizando uma variável threshold compartilhada para todas as ReLUs.

Você poderia simplesmente criar essa variável primeiro e depois passá-la para a função relu():

def relu(X, threshold):

with tf.name_scope("relu"):

[...]

return tf.maximum(z, threshold, name="max")

threshold = tf.Variable(0.0, name="threshold")