Create A 3d Matix With Python Tensorflow