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]