创建一个占位符，该占位符的形状是其他形状的函数

Question

假设我们有一个张量流占位符，如下所示：

x = tf.placeholder(tf.float32, (2, 2, 3 ..., 1))

我想创建另一个张量y，其形状为x，除了第一和第二维是x的三倍。

y = tf.placeholder(tf.float32, (6, 6, 3, ..., 1))

x的形状不是预定义的，所以我想做的事情如下：

y = tf.placeholder(tf.float32, (x.shape[0]* 3, x.shape[1] * 3, remaining_are_the_same_as_x_shape))

您能建议我如何在张量流中做到这一点吗？

Answer 1

那呢？

x = tf.placeholder(tf.float32, (2, 2, 3 , 1))

shape = x.get_shape().as_list()
shape[0] = shape[0] * 3
shape[1] = shape[1] * 3

y = tf.placeholder(tf.float32, shape=shape)
shape = y.get_shape().as_list()
print(shape)

  

[6、6、3、1]