fork download
  1. import tensorflow as tf
  2. import numpy as np
  3. import sys
  4. print(sys.version)
  5.  
  6. x = [[10, 20, 50, 60, 30, 40, 70, 80],
  7. [110, 120, 150, 160, 130, 140, 170, 180]];
  8.  
  9. gfg, geeks = tf.split(x, num_or_size_splits=2, axis=-1);
  10.  
  11. sess = tf.Session()
  12. gfg_v,geeks_v = sess.run([gfg, geeks])
  13.  
  14. print gfg_v;
  15. print gfg_v.shape;
  16. print geeks_v;
  17. print geeks_v.shape;
  18.  
  19. output_size=np.prod(geeks_v.shape) * 2
  20. print output_size
Success #stdin #stdout 1.44s 204700KB
stdin
Standard input is empty
stdout
2.7.16 (default, Apr  6 2019, 01:42:57) 
[GCC 8.3.0]
[[ 10  20  50  60]
 [110 120 150 160]]
(2, 4)
[[ 30  40  70  80]
 [130 140 170 180]]
(2, 4)
16