tf.split()

前言：tensorflow版本1.4.0

定义：将一个张量分为几个分张量。

格式：

split(
    value,    #输入张量
    num_or_size_splits,  #每个分割后的张量的尺寸
    axis=0,  #被分张量的分割标准//当axis=0时按行分，当axis=1时，按列分。
    num=None,
    name='split'
)

代码实现：

import tensorflow as tf
value = [[1,2,3],[4,5,6]]

split0, split1 = tf.split(value, [1, 1], 0)
with tf.Session() as sess:
    print(sess.run(split0))
    print(sess.run(split1))

结果：

[[1 2 3]]

[[4 5 6]]

import tensorflow as tf
value = [[1,2,3],[4,5,6]]

split0, split1,split2 = tf.split(value, [1, 2,0], 1)
with tf.Session() as sess:
    print(sess.run(split0))
    print(sess.run(split1))
    print(sess.run(split2))

结果：

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