tensorflow 的tf.split函数的用法

将张量进行切分

tf.split(
    value,
    num_or_size_splits,
    axis=0,
    num=None,
    name='split'
)

value:　　待切分的张量

num_or_size_splits:　　切分的个数

axis： 沿着哪个维度切分