使用与未知批量大小

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)设置对于批量动态形状。