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=0);
  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.11s 205424KB
stdin
Standard input is empty
stdout
2.7.16 (default, Apr  6 2019, 01:42:57) 
[GCC 8.3.0]
[[10 20 50 60 30 40 70 80]]
(1, 8)
[[110 120 150 160 130 140 170 180]]
(1, 8)
16