Tensorflow 2x

shape