python - TensorFlow 中的 GRUCell : TypeError got multiple values for keyword argument 'num_units'

Question

所以我有这个非常简单的玩具代码:

import tensorflow as tf

x = tf.placeholder(tf.int32, [None, 10])

def new_network(x):
    return  tf.nn.rnn_cell.GRUCell(x, num_units=100)

pred = new_network(x)

无论我做什么，我都会收到以下错误

TypeError: __init__() got multiple values for keyword argument 'num_units'

我使用的是 TensorFlow 1.3.0 版本。

这与this pulled issue有关吗？ ？

Answer 1

tf.nn.rnn_cell.GRUCell使用 num_units 和激活等进行初始化，但不使用输入进行初始化。当您使用 __call__ 方法

时会传递输入

GRUCell.__init__(
    num_units,
    activation=None,
    reuse=None,
    kernel_initializer=None,
    bias_initializer=None
)

state = tf.placeholder(tf.int32, [None, state_size])
def new_network(x):
    gru_cell = tf.nn.rnn_cell.GRUCell(num_units=100)
    y = gru_cell(x, state)
pred = new_network(x)