python - 指定动态形状作为输入参数

Question

对于需要形状参数的 tensorflow 函数，我们如何提供动态形状作为输入参数之一？

例如，假设我有一个占位符

x = tf.placeholder(tf.float32, [None, 3])

现在我想创建一个与 x 第一个维度形状相同的一维随机张量。

r = tf.random_uniform([x.get_shape()[0]])

这不起作用，因为 x.get_shape()[0] 将返回静态形状“None”。有没有办法根据x的动态形状为r分配动态形状？

Answer 1

您可以使用 tf.shape() 获取张量的动态形状op，并将其用作 tf.random_uniform() 的参数:

x = tf.placeholder(tf.float32, [None, 3])
r = tf.random_uniform(tf.shape(x)[0:1])

(请注意，tf.shape(x) 的切片，即 [0:1] 必须指定一个范围而不是单个元素，因为 tf.random_uniform() 需要一个维度向量。)