如何从Tensorflow中的张量中获取特定行？

import tensorflow as tf

# 创建一个二维张量
tensor = tf.constant([[1, 2, 3],
                      [4, 5, 6],
                      [7, 8, 9]])

# 获取第一行数据
row_1 = tensor[0]
print(row_1)  # 输出: [1 2 3]

# 获取第二和第三行数据
row_2_3 = tensor[1:3]
print(row_2_3)  # 输出: [[4 5 6]
                #       [7 8 9]]

# 获取第一和第三行数据
mask = tf.constant([True, False, True])
selected_rows = tensor[mask]
print(selected_rows)  # 输出: [[1 2 3]
                      #       [7 8 9]]