tensorflow的基本应用

1  tf.split()

  

f.split(dimension, num_split, input)：dimension的意思就是输入张量的哪一个维度，如果是0就表示对第0维度进行切割。num_split就是切割的数量，如果是2就表示输入张量被切成2份，每一份是一个列表。

例如：

1. import tensorflow as tf;
2. import numpy as np;
4. A = [[ 1, 2, 3],[ 4, 5, 6]]
5. x = tf.split( 1, 3, A)
7. with tf.Session() as sess:
8. c = sess.run(x)
9. for ele in c:
10. print ele

输出：

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