使用與未知批量大小

Question

tf.get_variable（...）考慮到我有一些輸入批次x與形狀(batch_size, n_features)我會怎麼用tf.get_variable(...)與形狀(batch_size)創建另一個矩陣如果不是提前知道。

例如，我能夠做到：

b = tf.zeros(shape = (tf.shape(x)[0]))

但我不能這樣做：

b = tf.get_variable("b", shape = (tf.shape(x)[0]), initializer = tf.constant_initializer(0.0))

Answer 1

您可以使用tf.Variable(dims=[tf.shape(x)[0]], validate_shape=False)設置對於批量動態形狀。