tensorflow中np.insert的替代方案

import tensorflow as tf

def insert_value(tensor, value, axis, index):
    before = tf.slice(tensor, [0] * tensor.shape.ndims, [index] + [-1] * (tensor.shape.ndims - 1))
    after = tf.slice(tensor, [index] + [-1] * (tensor.shape.ndims - 1), [-1] * tensor.shape.ndims)
    return tf.concat([before, value, after], axis=axis)

# 示例用法
tensor = tf.constant([[1, 2, 3], [4, 5, 6]])
value = tf.constant([7, 8, 9])
axis = 0
index = 1

result = insert_value(tensor, value, axis, index)
print(result.numpy())