在不使用tf.RaggedTensor的情况下从tensorflow的张量中删除某些行

import tensorflow as tf

def remove_rows(tensor, indices):
    mask = tf.reduce_all(tf.math.not_equal(tf.range(tensor.shape[0])[:, tf.newaxis], indices), axis=1)
    return tf.boolean_mask(tensor, mask)

# 示例数据
tensor = tf.constant([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])
indices = tf.constant([1, 3])

# 删除指定行
new_tensor = remove_rows(tensor, indices)

print(new_tensor)

tf.Tensor(
[[ 1  2  3]
 [ 7  8  9]], shape=(2, 3), dtype=int32)