tensorflow中tf.split()和tf.concat()的使用

import tensorflow as tf

a = [[[1,1,1], [2,2,2], [3,3,3]], [[4,4,4], [5,5,5], [6,6,6]]]

print (a)

b = tf.split(a, 3, 2)

print (b)

c = tf.concat(b, 0)

print (c)

'''
张量a的shape为(2,3,3)。tf.split(a, 3, 2)，意味着直接在axis=2这个维度上，
对张量a平等划分为3个小张量。
[<tf.Tensor: shape=(2, 3, 1), dtype=int32, numpy=
array([[[1],
        [2],
        [3]],

       [[4],
        [5],
        [6]]])>, <tf.Tensor: shape=(2, 3, 1), dtype=int32, numpy=
array([[[1],
        [2],
        [3]],

       [[4],
        [5],
        [6]]])>, <tf.Tensor: shape=(2, 3, 1), dtype=int32, numpy=
array([[[1],
        [2],
        [3]],

       [[4],
        [5],
        [6]]])>]

tf.concat(b, 0)，意味着在axis=0这个维度上，对张量b进行拼接。
tf.Tensor(
[[[1]
  [2]
  [3]]

 [[4]
  [5]
  [6]]

 [[1]
  [2]
  [3]]

 [[4]
  [5]
  [6]]

 [[1]
  [2]
  [3]]

 [[4]
  [5]
  [6]]], shape=(6, 3, 1), dtype=int32)

tf.concat(b, 1)：

tf.Tensor(
[[[1]
  [2]
  [3]
  [1]
  [2]
  [3]
  [1]
  [2]
  [3]]

 [[4]
  [5]
  [6]
  [4]
  [5]
  [6]
  [4]
  [5]
  [6]]], shape=(2, 9, 1), dtype=int32)
'''