在TF / Numpy中进行此连接的优雅方法？

import tensorflow as tf

# 创建两个张量
tensor1 = tf.constant([[1, 2], [3, 4]])
tensor2 = tf.constant([[5, 6]])

# 在轴0上连接
result_axis0 = tf.concat([tensor1, tensor2], axis=0)
print("Concatenated along axis 0:\n", result_axis0.numpy())

# 在轴1上连接
result_axis1 = tf.concat([tensor1, tensor2], axis=1)
print("Concatenated along axis 1:\n", result_axis1.numpy())

import numpy as np

# 创建两个数组
array1 = np.array([[1, 2], [3, 4]])
array2 = np.array([[5, 6]])

# 在轴0上连接
result_axis0 = np.concatenate((array1, array2), axis=0)
print("Concatenated along axis 0:\n", result_axis0)

# 在轴1上连接
result_axis1 = np.concatenate((array1, array2), axis=1)
print("Concatenated along axis 1:\n", result_axis1)