具有动态形状TensorFlow的变量

Question

我需要在TensorFlow中创建一个矩阵来存储一些值.诀窍是矩阵必须支持动态形状.

我试图做同样的事情,我会做numpy:

myVar = tf.Variable(tf.zeros((x,y), validate_shape=False)

在哪里x=(?)和y=2.但这不起作用,因为零不支持'部分已知的TensorShape',所以,我应该如何在TensorFlow中执行此操作？

Answer 1

1）您可以使用tf.fill(dims, value=0.0)它来处理动态形状。

2）您可以使用变量维度的占位符，例如：

m = tf.placeholder(tf.int32, shape=[])
x = tf.zeros(shape=[m])

with tf.Session() as sess:
    print(sess.run(x, feed_dict={m: 5}))