a = tf.constant([[1, 2], [3, 4]])
a_as_vector = tf.reshape(a, [-1])
zero_padding = tf.zeros([2 * 3] - tf.shape(a_as_vector), dtype=a.dtype)
a_padded = tf.concat(0, [a_as_vector, zero_padding])
result = tf.reshape(a_padded, [2, 3])
Изменение формы тензора Tensorflow и дополнение нулями