Создание заполнителя, имеющего форму, которая является функцией другой формы

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))

Не могли бы вы посоветовать мне, как это сделать в tenorflow?

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]