Reshape Extra

07 May 2019

import tensorflow as tf

import os

# Just disables the warning, doesn't enable AVX/FMA
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

a = tf.constant([1, 2, 3, 4, 5, 6, 7, 9])

b = tf.reshape(a, [2, 4])

<tf.Tensor 'Const_1:0' shape=(8,) dtype=int32>

<tf.Tensor 'Reshape_2:0' shape=(2, 4) dtype=int32>

with tf.Session() as sess:
    b_val = sess.run(b)
    print(b_val)

[[1 2 3 4]
 [5 6 7 9]]