tf.concat()和np.concatenate()用法的一个坑

tf.concat()是tensorflow中用来拼接张量的函数，用法:
tf.concat([tensor1, tensor2, tensor3,…], axis)
关键来了 如果只给出了一个张量 即：
tf.concat(tensor1, axis)
这种情况下不会对tensor1做任何操作，依然返回tensor1

再来看np.concatenate(), 用法：
np.concatenate((a1, a2, …), axis=0)
同样如果只给一个数组 即：
np.concatenate(a1, axis=0)
函数会把a1拆开，再进行拼接，
举个例子：
import numpy as np

a = np.array([[1, 2], [3, 4]])
c = np.concatenate(a, axis=0)
print( c )

结果是

[1, 2, 3, 4]