如何沿列(垂直)对2D矩阵进行切片并在tensorflow中创建3D矩阵？

tf.slice(input_, begin, size, name=None)

import tensorflow as tf

# 创建一个2D矩阵
matrix_2d = tf.constant([[1, 2, 3],
                         [4, 5, 6],
                         [7, 8, 9]])

# 沿列（垂直）对矩阵进行切片
slice_2d = tf.slice(matrix_2d, [0, 1], [3, 1])

# 在TensorFlow中创建3D矩阵
matrix_3d = tf.expand_dims(slice_2d, axis=2)

# 打印结果
with tf.Session() as sess:
    result = sess.run(matrix_3d)
    print(result)

[[[2]
  [5]
  [8]]]