使用具有未知批量大小的tf.get_variable（...）

Question

鉴于我有一些形状为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)为批量大小设置动态形状。