我需要在 TensorFlow 中创建一个矩阵来存储一些值。诀窍是矩阵必须支持动态形状。
我正在尝试做与 numpy 中相同的事情:
myVar = tf.Variable(tf.zeros((x,y), validate_shape=False)
where x=(?)
and y=2
。但这不起作用,因为 Zeros 不支持“部分已知的 TensorShape”,所以,我应该如何在 TensorFlow 中执行此操作?
1)你可以使用tf.fill(dims, value=0.0) https://www.tensorflow.org/api_docs/python/tf/fill适用于动态形状。
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}))
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)